Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Daggregate_ops.cc57 xla::Shape sum_shape; in Compile() local
/external/ComputeLibrary/src/core/CL/kernels/
DCLL2NormalizeLayerKernel.cpp60 TensorShape sum_shape = input->tensor_shape(); in validate_arguments() local
/external/ComputeLibrary/src/core/NEON/kernels/
DNEL2NormalizeLayerKernel.cpp145 TensorShape sum_shape = input->tensor_shape(); in validate_arguments() local
/external/tensorflow/tensorflow/core/kernels/
Dsparse_conditional_accumulator.h239 TensorShape sum_shape = grad_val->shape(); in AddToAccumGradFunction() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc189 auto sum_shape = builder.GetShape(sum).ConsumeValueOrDie(); in XLA_TEST_P() local