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 | 1797 Value zero_tensor = in matchAndRewrite() local
1801 ValueRange{zero_tensor}); in matchAndRewrite()
1865 Value zero_tensor = in matchAndRewrite() local
1870 /*outputBuffers=*/ValueRange{zero_tensor}); in matchAndRewrite()
2103 Value zero_tensor = in matchAndRewrite() local
2114 ValueRange{zero_tensor}, dilations, strides); in matchAndRewrite()
2120 ValueRange{zero_tensor}, dilations, strides); in matchAndRewrite()
2126 ValueRange{zero_tensor}, dilations, strides); in matchAndRewrite()
2228 Value zero_tensor = in matchAndRewrite() local
2235 ValueRange{zero_tensor}, rhs_dilation, window_strides); in matchAndRewrite()
[all …]
|