Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dsparsify_gather.cc392 Tensor zero_tensor(data_type, TensorShape({})); in SparsifyGatherInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dquantize_training.cc335 Tensor zero_tensor(DT_INT32, TensorShape()); in MakeReductionAxes() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dops.cpp69 bool zero_tensor = false; in repeat_out() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp1517 bool zero_tensor = false; in repeat() local