Home
last modified time | relevance | path

Searched defs:zero_tensor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dsparsify_gather.cc392 Tensor zero_tensor(data_type, TensorShape({})); in SparsifyGatherInternal() local
/external/tensorflow/tensorflow/core/common_runtime/
Dquantize_training.cc335 Tensor zero_tensor(DT_INT32, TensorShape()); in MakeReductionAxes() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc1147 Value zero_tensor = in matchAndRewrite() local
1235 Value zero_tensor = in matchAndRewrite() local