Cross Reference: /external/tensorflow/tensorflow/compiler/xla/service/shape

Lines Matching refs:offset_shape
1181     const Shape& offset_shape, int64 feature_index) {
1185       offset_shape, "offset input of batch norm training"));
1191   TF_RET_CHECK(ShapeUtil::ValidateShapeWithOptionalLayout(offset_shape) ==
1218   if (ShapeUtil::Rank(offset_shape) != 1) {
1222         ShapeUtil::Rank(offset_shape));
1239   if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(offset_shape,
1245         PrimitiveType_Name(offset_shape.element_type()).c_str(),
1263   if (ShapeUtil::GetDimension(offset_shape, 0) != feature_count) {
1268         ShapeUtil::GetDimension(offset_shape, 0), feature_count);
1286     const Shape& offset_shape, const Shape& mean_shape,
1291       offset_shape, "offset input of batch norm inference"));
1297   TF_RET_CHECK(ShapeUtil::ValidateShapeWithOptionalLayout(offset_shape) ==
1328   if (ShapeUtil::Rank(offset_shape) != 1) {
1332         ShapeUtil::Rank(offset_shape));
1349   if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(offset_shape,
1356         PrimitiveType_Name(offset_shape.element_type()).c_str(),
1397   if (ShapeUtil::GetDimension(offset_shape, 0) != feature_count) {
1402         ShapeUtil::GetDimension(offset_shape, 0), feature_count);
OpenGrok
