Searched refs:scalar_zero (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 150 Value scalar_zero = CreateI32SplatConst(loc, rewriter, {}, 0); in CreateSliceOpForTensorList() local
153 loc, vector_type, start_index, scalar_zero); in CreateSliceOpForTensorList()
155 loc, position_type, scalar_zero, in CreateSliceOpForTensorList()
161 rewriter->create<TF::ExpandDimsOp>(loc, vector_type, size, scalar_zero); in CreateSliceOpForTensorList()
164 loc, position_type, scalar_zero, in CreateSliceOpForTensorList()
272 Value scalar_zero = CreateI32SplatConst(loc, &rewriter, {}, 0); in matchAndRewrite() local
281 loc, RankedTensorType::get({1}, shape_dtype), item_rank, scalar_zero); in matchAndRewrite()
288 /*start_index=*/scalar_zero, in matchAndRewrite()
302 op.getLoc(), unranked_tensor, item, scalar_zero); in matchAndRewrite()
306 op, input.getType(), scalar_zero, in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_test.cc | 619 auto scalar_zero = LiteralUtil::CreateR0<float>(0.0f); in TEST_F() local
621 EXPECT_TRUE(scalar_zero.IsZero({})); in TEST_F()
|