Searched refs:zero_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather.cc | 392 Tensor zero_tensor(data_type, TensorShape({})); in SparsifyGatherInternal() local
393 zero_tensor.flat<float>()(0) = 0.0; in SparsifyGatherInternal()
395 CreateConstNode(zero_tensor, StrCat(gather_node.name(), "/Const"), in SparsifyGatherInternal()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | quantize_training.cc | 335 Tensor zero_tensor(DT_INT32, TensorShape()); in MakeReductionAxes() local
336 zero_tensor.flat<int32>()(0) = 0; in MakeReductionAxes()
340 .Attr("value", zero_tensor) in MakeReductionAxes()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 1147 Value zero_tensor = in matchAndRewrite() local
1154 ValueRange{adaptor.lhs(), adaptor.rhs()}, ValueRange{zero_tensor}); in matchAndRewrite()
1160 ValueRange{adaptor.lhs(), adaptor.rhs()}, ValueRange{zero_tensor}); in matchAndRewrite()
1166 ValueRange{adaptor.lhs(), adaptor.rhs()}, ValueRange{zero_tensor}); in matchAndRewrite()
1235 Value zero_tensor = in matchAndRewrite() local
1240 /*outputBuffers=*/ValueRange{zero_tensor}); in matchAndRewrite()
|