Searched refs:offset_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander_test.cc | 59 Shape offset_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
69 HloInstruction::CreateParameter(2, offset_shape, "offset")); in TEST_F()
72 ShapeUtil::MakeTupleShape({input_shape, scale_shape, offset_shape}), in TEST_F()
|
| D | shape_inference.cc | 1168 const Shape& offset_shape, int64 feature_index) { in InferBatchNormTrainingShape() argument
1172 ExpectArray(offset_shape, "offset input of batch norm training")); in InferBatchNormTrainingShape()
1178 TF_RET_CHECK(ShapeUtil::ValidateShapeWithOptionalLayout(offset_shape) == in InferBatchNormTrainingShape()
1205 if (offset_shape.rank() != 1) { in InferBatchNormTrainingShape()
1209 offset_shape.rank()); in InferBatchNormTrainingShape()
1226 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(offset_shape, in InferBatchNormTrainingShape()
1232 PrimitiveType_Name(offset_shape.element_type()), in InferBatchNormTrainingShape()
1250 if (ShapeUtil::GetDimension(offset_shape, 0) != feature_count) { in InferBatchNormTrainingShape()
1255 ShapeUtil::GetDimension(offset_shape, 0), feature_count); in InferBatchNormTrainingShape()
1273 const Shape& offset_shape, const Shape& mean_shape, in InferBatchNormInferenceShape() argument
[all …]
|
| D | shape_inference.h | 90 const Shape& offset_shape,
97 const Shape& offset_shape, const Shape& mean_shape,
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_batch_normalization.cc | 95 const auto& offset_shape = offset_array.shape(); in Run() local
103 CHECK(mean_shape.dims() == offset_shape.dims()); in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2053 TF_ASSIGN_OR_RETURN(const Shape& offset_shape, GetShape(offset)); in BatchNormTraining()
2057 operand_shape, scale_shape, offset_shape, feature_index)); in BatchNormTraining()
2077 TF_ASSIGN_OR_RETURN(const Shape& offset_shape, GetShape(offset)); in BatchNormInference()
2082 operand_shape, scale_shape, offset_shape, mean_shape, in BatchNormInference()
|