| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | symbolic_shape_analysis.cpp | 104 if (auto val = sa.asConstantInt()) { in operator <<() local
236 void replaceWithIValue(Value* v, const IValue& val) { in replaceWithIValue()
1132 const std::vector<SSAInput>& inputs) { in calculateSymbolicShapesOnOp()
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | NCCLUtils.cpp | 162 const char* val = getenv("TORCH_NCCL_NONBLOCKING_TIMEOUT"); in _parse_nccl_nonblocking_timeout() local
380 const std::vector<at::Tensor>& inputs, in record()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | register_prim_ops_fulljit.cpp | 234 auto inputs = pop(stack).toTensorList(); in __anon1bf66a481902() local
343 auto val = pop(stack).toInt(); in __anon1bf66a482002() local
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | checkpoint.py | 1167 def __init__(self, val): argument
1500 def recompute_fn(*inputs):
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_decomp.py | 103 outputs, inputs, grad_outputs=None, retain_graph=False, create_graph=True argument
139 def _as_tuple(val): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/ir/ |
| H A D | ir.cpp | 1698 ArrayRef<Value*> inputs, in create()
1947 std::vector<Value*> inputs = {fn_constant}; in insertFunctionCall() local
1987 const IValue& val, in insertConstant()
2219 ArrayRef<Value*> inputs, in insertGraph()
2244 ArrayRef<Value*> inputs) { in insertGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | sharding_util_ops_test.cc | 794 int32_t val = 0; in CreateConcatTensorGraph() local
796 std::vector<NodeBuilder::NodeOut> inputs; in CreateConcatTensorGraph() local
845 int32_t val = 0; in CreateConcatResourceGraph() local
846 std::vector<NodeBuilder::NodeOut> inputs; in CreateConcatResourceGraph() local
|
| /aosp_15_r20/external/armnn/src/armnnOnnxParser/ |
| H A D | OnnxParser.cpp | 411 int val = CHECKED_INT32(targetShapeTensor[i]); in ComputeReshapeInfo() local
1405 std::pair<std::string, std::string> inputs = std::make_pair(input0, input1); in AddPrepareBroadcast() local
1535 auto inputs = AddPrepareBroadcast(node.input(0), node.input(1)); in ParseAdd() local
2230 … int val = CHECKED_INT32(m_TensorsInfo[node.input(1)].m_tensor->int64_data(static_cast<int>(i))); in ParseReshape() local
|
| /aosp_15_r20/prebuilts/go/linux-x86/src/cmd/compile/internal/ssa/ |
| D | op.go | 66 inputs []inputInfo member
425 func makeValAndOff(val, off int32) ValAndOff { argument
|
| /aosp_15_r20/external/trace-cmd/tracecmd/ |
| H A D | trace-read.c | 651 static void add_wakeup(unsigned int val, unsigned long long start) in add_wakeup()
683 static void add_sched(unsigned int val, unsigned long long end, int rt) in add_sched()
734 unsigned long long val; in process_wakeup() local
1464 char *val = NULL; in process_plugin_option() local
1558 struct input_files *inputs; in trace_report() local
|
| /aosp_15_r20/external/linux-kselftest/tools/testing/selftests/kvm/include/x86_64/ |
| H A D | processor.h | 379 static inline void set_cr4(uint64_t val) in set_cr4()
442 static inline void set_xmm(int n, unsigned long val) in set_xmm()
811 #define kvm_asm_safe(insn, inputs...) \ argument
822 static inline uint8_t rdmsr_safe(uint32_t msr, uint64_t *val) in rdmsr_safe()
836 static inline uint8_t wrmsr_safe(uint32_t msr, uint64_t val) in wrmsr_safe()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | function_test.py | 715 def a_times_b(inputs): argument
997 def a_times_b(inputs): argument
1267 def test_function(val): argument
1739 def my_op(inputs): argument
3229 def call(self, inputs, training=True): argument
3647 def function_callback(f, name, graph, inputs, outputs): argument
3674 def function_callback(f, name, graph, inputs, outputs): argument
3699 def function_callback_1(f, name, graph, inputs, outputs): argument
3704 def function_callback_2(f, name, graph, inputs, outputs): argument
4234 def summing_rnn(inputs): argument
[all …]
|
| /aosp_15_r20/external/mesa3d/src/panfrost/vulkan/ |
| H A D | panvk_vX_shader.c | 79 nir_def *val = NULL; in panvk_lower_sysvals() local
768 struct panfrost_compile_inputs inputs = { in panvk_compile_shader() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/ |
| D | NeuralNetworks.cpp | 1183 const ModelBuilder* val = reinterpret_cast<const ModelBuilder*>(value); in ANeuralNetworksModel_setOperandValueFromModel() local
1190 const uint32_t* inputs, uint32_t outputCount, in ANeuralNetworksModel_addOperation()
1215 const uint32_t* inputs, uint32_t outputCount, in ANeuralNetworksModel_identifyInputsAndOutputs()
|
| /aosp_15_r20/external/mesa3d/src/gallium/auxiliary/nir/ |
| H A D | tgsi_to_nir.c | 75 nir_variable **inputs; member
2165 nir_def *val = zero; in ttn_add_output_stores() local
2490 nir_def *val = zero; in lower_clipdistance_to_array() local
|
| /aosp_15_r20/prebuilts/devtools/tools/lib/ |
| HD | ant-tasks.jar | META-INF/
META-INF/MANIFEST.MF
com/
com/android/
com/ ... |
| /aosp_15_r20/external/mesa3d/src/gallium/auxiliary/draw/ |
| H A D | draw_llvm.c | 646 LLVMValueRef *inputs, in fetch_instanced()
704 LLVMValueRef *inputs, in fetch_vector()
894 LLVMValueRef val, cliptmp; in draw_store_aos_array() local
1832 LLVMValueRef inputs[PIPE_MAX_SHADER_INPUTS][TGSI_NUM_CHANNELS]; in draw_llvm_generate() local
2868 LLVMValueRef val = LLVMBuildExtractElement(builder, value, idx, ""); in draw_tcs_llvm_emit_store_output() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | batch_kernels.cc | 102 const char* val = std::getenv("TF_NUM_BATCH_THREADS"); in NumBatchThreadsFromEnvironmentWithDefault() local
204 const BatchTask& last_task, absl::Span<const Tensor> inputs, in ProcessFuncBatchImpl()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | shape_inference.cc | 485 ValuePortInputs* inputs) { in ComputeInputsRequiredForOutput()
572 ValuePortInputs* inputs) { in ComputeInputsRequiredForOutput()
1855 SmallVector<ValuePort, 4> inputs; in ComputeOutputAsShape() local
1884 int64_t val = (*dea.getValues<APInt>().begin()).getSExtValue(); in ComputeOutputAsShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | functionalize_cond.cc | 134 Status GetSwitchValue(const Node& switch_node, OutputTensor* val) { in GetSwitchValue()
774 std::vector<NodeDefBuilder::NodeOut> inputs; in BuildIfNode() local
1265 OutputTensor val; in RemoveRedundantSwitch() local
|
| /aosp_15_r20/external/deqp/modules/glshared/ |
| H A D | glsVertexArrayTests.cpp | 378 static inline void alignmentSafeAssignment(char *dst, T val) in alignmentSafeAssignment()
775 void ContextShaderProgram::shadeVertices(const rr::VertexAttrib *inputs, rr::VertexPacket *const *p… in shadeVertices()
1718 T val = getRandom<T>(rnd, min, max); in createPerQuads() local
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | utils.py | 632 def make_cell(val=None): argument
762 def create(scope, name, val): argument
1615 def __exit__(self, typ, val, traceback): argument
2710 def flatten_graph_inputs(gm: torch.fx.GraphModule, inputs, compile_gm): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | testlib.cc | 116 Node* Assign(Graph* g, Node* var, Node* val) { in Assign()
345 std::vector<NodeBuilder::NodeOut> inputs; in Merge() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/r600/sfn/ |
| H A D | sfn_instr_tex.cpp | 49 TexInstr::set_offset(unsigned index, int32_t val) in set_offset()
704 TexInstr::prepare_source(nir_tex_instr *tex, const Inputs& inputs, Shader& shader) in prepare_source()
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | test_export.py | 296 self, model, inputs, specs, passing_shapes, failing_shapes, test_serdes=False argument
828 def set_tensor(self, i, val): argument
857 def set_tensor(self, i, val): argument
975 def _tuple_contains(_tuple, val): argument
3003 def forward(self, inputs): argument
3024 def forward(self, inputs): argument
3060 def forward(self, inputs): argument
3231 def helper(model, inputs, dynamic_shapes): argument
3372 def forward(self, inputs): argument
5075 def forward_hook(module: torch.nn.Module, inputs, output) -> torch.Tensor: argument
|