| /external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/template_writer/ |
| D | GpuKernelVariableTable.h | 54 struct TensorVariable struct
57 TensorVariable() = default;
58 TensorVariable(const TensorVariable &) = default;
59 TensorVariable &operator=(const TensorVariable &) = default; argument
68 using VariableList = std::vector<TensorVariable>; argument
85 TensorVariable get_variable(const ITensorInfo *tensor) const;
96 std::map<ITensorInfo::Id, TensorVariable> _vars{};
105 TagVal(const GpuKernelVariableTable::TensorVariable &var);
|
| D | GpuKernelVariableTable.cpp | 67 TensorVariable var{ tensor->id(), uniq_name, argument_info }; in declare_variable()
73 GpuKernelVariableTable::TensorVariable GpuKernelVariableTable::get_variable(const ITensorInfo *tens… in get_variable()
93 TagVal::TagVal(const GpuKernelVariableTable::TensorVariable &var) in TagVal()
|
| /external/pytorch/torch/_dynamo/variables/ |
| D | torch.py | 365 TensorVariable,
413 if isinstance(arg, TensorVariable) or (
428 if isinstance(input_arg, TensorVariable) and input_arg.dtype is not None:
438 if isinstance(input, TensorVariable) and input.size is not None:
440 elif isinstance(input, TensorVariable):
446 assert isinstance(input, TensorVariable)
513 target_cls=TensorVariable,
538 tensor, TensorVariable
596 if len(args) == 2 and isinstance(args[0], TensorVariable) and not kwargs:
757 if isinstance(x, (TensorVariable, SymNodeVariable)):
[all …]
|
| D | builtin.py | 68 TensorVariable,
489 TensorVariable,
576 (TensorVariable, VariableTracker),
580 (VariableTracker, TensorVariable),
667 any_tensor = any_tensor or isinstance(arg, variables.TensorVariable)
675 any_tensor = any_tensor or issubclass(arg_type, variables.TensorVariable)
682 if isinstance(i, variables.TensorVariable):
733 and not issubclass(arg_types[0], variables.TensorVariable)
896 type(arg) == variables.TensorVariable for arg in args
1000 if isinstance(arg, (SymNodeVariable, variables.TensorVariable)):
[all …]
|
| D | higher_order_ops.py | 448 from . import TensorVariable
450 if not only_consist_of(output, TensorVariable, allow_none=True):
577 from . import ListVariable, TensorVariable
599 if type(args[0]) not in (ConstantVariable, TensorVariable, SymNodeVariable):
612 if not only_consist_of(args[3], (TensorVariable,)):
652 if not only_consist_of(ret_val, (TensorVariable,)):
679 isinstance(var, TensorVariable) for var in tensor_vars1 + tensor_vars2
790 from . import TensorVariable
821 if not only_consist_of(args[2], (TensorVariable,)):
847 if not isinstance(cond_r, TensorVariable):
[all …]
|
| D | tensor.py | 92 class TensorVariable(VariableTracker): class
814 if isinstance(v, TensorVariable):
823 and isinstance(value, TensorVariable)
1144 class NumpyNdarrayVariable(TensorVariable):
1256 class UnspecializedPythonVariable(TensorVariable):
1264 *TensorVariable._nonvar_fields,
1284 class FakeItemVariable(TensorVariable):
1290 *TensorVariable._nonvar_fields,
1314 if len(args) == 1 and isinstance(args[0], TensorVariable):
1340 from_tensor: TensorVariable, argument
|
| D | __init__.py | 100 TensorVariable,
|
| D | torch_function.py | 19 from .tensor import TensorSubclassVariable, TensorVariable
265 class TensorWithTFOverrideVariable(TensorVariable):
|
| D | lists.py | 120 from .tensor import TensorVariable
123 if isinstance(args[0], TensorVariable):
787 if isinstance(start, variables.TensorVariable) or isinstance(
788 stop, variables.TensorVariable
|
| D | nn_module.py | 754 isinstance(x, variables.TensorVariable)
925 if isinstance(param, variables.TensorVariable)
1019 isinstance(value, variables.TensorVariable)
|
| D | misc.py | 649 if isinstance(node, variables.TensorVariable):
847 isinstance(x, variables.TensorVariable) and x.requires_grad
1097 args[0], variables.TensorVariable
1434 variables.TensorVariable,
|
| D | builder.py | 200 TensorVariable,
391 TensorVariable,
1234 target_cls=TensorVariable,
2037 return wrap_fx_proxy_cls(target_cls=TensorVariable, **kwargs)
2145 in (TensorVariable, TensorWithTFOverrideVariable),
|
| D | dicts.py | 33 if isinstance(x, variables.TensorVariable):
86 if isinstance(self.vt, variables.TensorVariable):
|
| D | functions.py | 386 if isinstance(x, variables.TensorVariable):
|
| D | user_defined.py | 525 and all(isinstance(x, variables.TensorVariable) for x in args[0].items)
|
| /external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/template_writer/cl/ |
| D | ClTemplateWriter.h | 82 std::string write_argument_declaration(const GpuKernelVariableTable::TensorVariable &var) const;
|
| D | ClTemplateWriter.cpp | 243 …TemplateWriter::write_argument_declaration(const GpuKernelVariableTable::TensorVariable &var) const in write_argument_declaration()
|
| /external/pytorch/torch/_dynamo/ |
| D | codegen.py | 31 TensorVariable,
194 TensorVariable,
|
| D | side_effects.py | 339 other_variable, variables.TensorVariable
448 assert isinstance(tensor, variables.TensorVariable)
|
| D | symbolic_convert.py | 122 from .variables.tensor import supported_comparison_ops, SymNodeVariable, TensorVariable
431 if isinstance(value, TensorVariable):
479 isinstance(value, (TensorVariable)) and self.should_compile_partial_graph()
527 elif not isinstance(value, TensorVariable) and value.has_unpack_var_sequence(
1627 if isinstance(fn, GetAttrVariable) and isinstance(fn.obj, TensorVariable):
1631 argsvars, (ConstantVariable, TensorVariable)
1958 if isinstance(seq, TensorVariable):
1960 elif isinstance(seq, GetAttrVariable) and isinstance(seq.obj, TensorVariable):
3006 symbolic_result: Optional[TensorVariable]
|
| D | utils.py | 1208 TensorVariable,
1221 if check_tensor_identity and isinstance(search, TensorVariable):
1229 if isinstance(x, TensorVariable):
|
| D | output_graph.py | 104 TensorVariable,
1108 and all(isinstance(x, TensorVariable) for x in stack_values)
|
| /external/pytorch/docs/source/ |
| D | torch.compiler_dynamo_deepdive.rst | 271 `TensorVariable <https://github.com/pytorch/pytorch/blob/83c0763dda1f93c6cf552ba88260a0dc7a3ecb70/t…
300 TRACE LOAD_GLOBAL y [TorchInGraphFunctionVariable(<built-in method any>), TensorVariable()]
|
| D | torch.compiler_troubleshooting.rst | 633 Reason: generic_jump TensorVariable()
|
| D | torch.compiler_faq.rst | 251 Reason: generic_jump TensorVariable()
|