Searched defs:scale_shape (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/python/ops/ |
D | nn_fused_batchnorm_test.py | 58 scale_shape, argument 110 scale_shape, argument 183 scale_shape, argument 243 scale_shape, argument
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | batchnorm_expander_test.cc | 44 Shape scale_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local 78 Shape scale_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
|
D | shape_inference.cc | 1180 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormTrainingShape() 1285 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormInferenceShape() 1433 const Shape& operand_shape, const Shape& scale_shape, in InferBatchNormGradShape()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_layout_assignment_test.cc | 249 Shape scale_shape = in TEST_F() local
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | image_ops.cc | 206 const TensorShape& scale_shape = context->InputShape(1); in Compile() local
|