Searched defs:scale_shape (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_fused_batchnorm_test.py | 63 scale_shape, argument
144 scale_shape, argument
225 scale_shape, argument
288 scale_shape, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | batchnorm_expander_test.cc | 57 Shape scale_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
92 Shape scale_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
|
| H A D | shape_inference.cc | 1210 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormTrainingShape()
1315 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormInferenceShape()
1459 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormGradShape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | nn_grad_test.cc | 366 TensorShape scale_shape({shape.dim_size(channel_dim)}); in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | image_ops.cc | 233 const TensorShape& scale_shape = context->InputShape(1); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | remapper_test.cc | 1729 auto scale_shape = ops::Placeholder::Shape({128}); in TEST_F() local
1808 auto scale_shape = ops::Placeholder::Shape({128}); in TEST_F() local
|