Home
last modified time | relevance | path

Searched defs:sum_shape (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Daggregate_ops.cc57 xla::Shape sum_shape; in Compile() local
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLL2NormalizeLayerKernel.cpp58 TensorShape sum_shape = input->tensor_shape(); in validate_arguments() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEL2NormalizeLayerKernel.cpp121 TensorShape sum_shape = input->tensor_shape(); in validate_arguments() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_conditional_accumulator.h239 TensorShape sum_shape = grad_val->shape(); in AddToAccumGradFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc173 auto sum_shape = builder.GetShape(sum).value(); in XLA_TEST_F() local