Home
last modified time | relevance | path

Searched defs:scale_shape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander_test.cc58 Shape scale_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
93 Shape scale_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
Dshape_inference.cc1211 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormTrainingShape()
1316 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormInferenceShape()
1460 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormGradShape()
/external/tensorflow/tensorflow/python/ops/
Dnn_fused_batchnorm_test.py63 scale_shape, argument
144 scale_shape, argument
225 scale_shape, argument
288 scale_shape, argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment_test.cc262 Shape scale_shape = in TEST_F() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper_test.cc760 auto scale_shape = ops::Placeholder::Shape({128}); in TEST_F() local
839 auto scale_shape = ops::Placeholder::Shape({128}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_ops.cc226 const TensorShape& scale_shape = context->InputShape(1); in Compile() local