Home
last modified time | relevance | path

Searched full:_set_item (Results 1 – 12 of 12) sorted by relevance

/external/pytorch/
Dpt_ops.bzl222 "aten::_set_item.t",
387 "aten::_set_item.str",
405 "aten::_set_item.int",
423 "aten::_set_item.bool",
441 "aten::_set_item.float",
459 "aten::_set_item.Tensor",
/external/pytorch/test/mobile/model_test/
Dmodel_ops.yaml41 aten::_set_item.int: 7
42 aten::_set_item.str: 315
43 aten::_set_item.t: 34
Dcoverage.yaml41 - aten::_set_item.int
42 - aten::_set_item.str
43 - aten::_set_item.t
728 aten::_set_item.int: 7
729 aten::_set_item.str: 163
730 aten::_set_item.t: 8
/external/pytorch/test/mobile/
Dtest_lite_script_type.py74 # ('operators', (('aten::_set_item', 'str', 3),)),
/external/pytorch/torch/csrc/jit/passes/
Dremove_mutation.cpp77 if (n->kind() != aten::_set_item || in removableSetItem()
197 case aten::_set_item: { in RemoveListMutation()
/external/pytorch/torch/csrc/jit/runtime/
Dserialized_shape_function_registry.cpp187 _6 = torch._set_item(wrapped_dims, i, dim)
305 _5 = torch._set_item(out, dim0, torch.floordiv(_4, step))
2114 _20 = torch._set_item(result_size, dim0, cat_dim_size)
2262 _27 = torch._set_item(result_size, dim2, cat_dim_size)
2371 _11 = torch._set_item(order, normalized_dst[i3], _10)
2372 _12 = torch._set_item(src_dims, normalized_src[i3], -1)
2373 _13 = torch._set_item(dst_dims, normalized_dst[i3], -1)
2391 _19 = torch._set_item(order, destination_dims[i4], _18)
2485 _10 = torch._set_item(out, infer_dim6, torch.floordiv(numel, newsize))
2921 _4 = torch._set_item(result0, dim, k)
Dregister_prim_ops.cpp690 "aten::_set_item.t(t [](a!) l, int idx, t(b -> *) el) -> t[](a!)"),
1665 TORCH_SELECTIVE_SCHEMA("aten::_set_item." key_type "(Dict(" key_type \
/external/pytorch/torch/csrc/jit/passes/onnx/
Dremove_inplace_ops_for_onnx.cpp304 TORCH_INTERNAL_ASSERT(n->kind() == aten::_set_item); in PrepareSetItemForONNX()
305 // It seems the JIT does not always produce an output for _set_item. in PrepareSetItemForONNX()
844 } else if (nkind == aten::_set_item) { in convertInplaceOpsAndTrackAlias()
/external/pytorch/aten/src/ATen/core/
Dinterned_strings.h198 _(aten, _set_item) \
/external/pytorch/torch/onnx/
Dsymbolic_opset11.py472 @_onnx_symbolic("aten::_set_item")
473 def _set_item(g: jit_utils.GraphContext, tensor_list, i, v): function
/external/pytorch/torch/csrc/jit/frontend/
Dir_emitter.cpp1815 loc, *graph, aten::_set_item, {self, input_k, input_v}, {}); in emitDictComprehension()
2834 aten::_set_item, in emitAugAssignmentGeneric()
2962 // Dispatch to aten::_set_item to both select and assign in emitSubscriptAssign()
2984 std::make_shared<BuiltinFunction>(aten::_set_item, std::nullopt)) in emitSubscriptAssign()
3865 aten::_set_item, in emitApplySpecialFormForDict()
/external/pytorch/benchmarks/static_runtime/
Dtest_static_runtime.cc2647 %updated : int[] = aten::_set_item(%res, %idx, %value) in TEST()