Searched refs:zero_tensor (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | Repeat.mm | 48 bool zero_tensor = false;
51 zero_tensor = true;
58 if (zero_tensor || result.numel() == 0) {
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A 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()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | ops.cpp | 69 bool zero_tensor = false; in repeat_out() local
72 zero_tensor = true; in repeat_out()
79 if (zero_tensor) { in repeat_out()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorShape.cpp | 1517 bool zero_tensor = false; in repeat() local
1520 zero_tensor = true; in repeat()
1535 if (zero_tensor) { in repeat()
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 977 zero_tensor = False
980 zero_tensor = True
983 if zero_tensor:
|
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | __init__.py | 1339 zero_tensor = scalar_tensor(0, dtype=basic_div.dtype, device=basic_div.device)
1342 floor_div = where(ne(div, 0), floor_div, copysign(zero_tensor, basic_div))
|