/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
H A D | eager_service_impl_test.cc | 192 inputs, in BuildOperation() 221 inputs, in AddOperationToEnqueueRequest() 231 inputs, in AddOperationToRunComponentFunctionRequest() 413 AttrValue val; in TEST_F() local 515 AttrValue val; in TestFunction() local 604 AttrValue val; in TestComponentFunction() local 770 AttrValue val; in Init() local 877 std::vector<RemoteTensorHandle> inputs = {input}; in TEST_F() local 933 std::vector<Tensor> inputs = {*input_tensor}; in TEST_F() local 983 TestExecuteNodeArgs inputs( in TEST_F() local [all …]
|
/aosp_15_r20/external/cronet/android/tools/gn2bp/ |
H A D | gn_utils.py | 227 def sources(self, val): argument 231 def inputs(self): member in GnParser.Target 235 def inputs(self, val): argument 243 def outputs(self, val): argument 251 def args(self, val): argument 259 def response_file_contents(self, val): argument 275 def deps(self, val): argument 283 def rust_flags(self, val): argument
|
/aosp_15_r20/external/cronet/components/cronet/gn2bp/ |
H A D | gn_utils.py | 233 def sources(self, val): argument 237 def inputs(self): member in GnParser.Target 241 def inputs(self, val): argument 249 def outputs(self, val): argument 257 def args(self, val): argument 265 def response_file_contents(self, val): argument 281 def deps(self, val): argument
|
/aosp_15_r20/external/mesa3d/src/gallium/drivers/llvmpipe/ |
H A D | lp_rast_debug.c | 199 char val, in plot() 219 char val) in debug_rectangle() 252 char val) in debug_blit_tile() 254 const struct lp_rast_shader_inputs *inputs = arg.shade_tile; in debug_blit_tile() local 271 char val) in debug_shade_tile() 276 const struct lp_rast_shader_inputs *inputs = arg.shade_tile; in debug_shade_tile() local 294 char val) in debug_clear_tile() 308 char val) in debug_triangle() 377 char val = get_label(j); in do_debug_bin() local
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
H A D | kernel.cpp | 133 const at::ArrayRef<torch::jit::Value*>& inputs) { in pickDeviceType() 400 auto val = toIValue(v).value(); in constant() local 448 auto val = toIValue(v).value(); in toArg() local 570 auto inputs = v->node()->inputs(); in computeValue() local 713 if (auto val = to<LongImm>(tc.node())) { in tripCount() local 1421 auto val = toIValue(v).value(); in bindConstant() local 1593 auto _prefer_static_mem = [](const torch::jit::Value* val) { in deduceMemoryLayoutPolicy() 1911 const at::ArrayRef<IValue>& inputs, in getStaticOutputSizesAndStrides() 1957 const at::ArrayRef<IValue>& inputs, in prepareRunArgs() 2027 auto inputs = last(stack, nInputs_); in runKernel() local [all …]
|
H A D | lowerings.cpp | 59 at::Device device) { in nnc_lowerings_lazy_registration() 108 [](const std::vector<ArgValue>& inputs, \ in nnc_lowerings_lazy_registration() 155 [](const std::vector<ArgValue>& inputs, \ in nnc_lowerings_lazy_registration() 245 [](const std::vector<ArgValue>& inputs, in nnc_lowerings_lazy_registration() 264 [](const std::vector<ArgValue>& inputs, in nnc_lowerings_lazy_registration() 283 [](const std::vector<ArgValue>& inputs, in nnc_lowerings_lazy_registration() 302 [](const std::vector<ArgValue>& inputs, in nnc_lowerings_lazy_registration() 321 [](const std::vector<ArgValue>& inputs, in nnc_lowerings_lazy_registration() 340 [](const std::vector<ArgValue>& inputs, in nnc_lowerings_lazy_registration() 359 [](const std::vector<ArgValue>& inputs, in nnc_lowerings_lazy_registration() [all …]
|
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/instancenorm/generic/neon/ |
H A D | impl.cpp | 34 void vector_float_sum(AccType &result, AccType &result_square, const InputType &inputs) in vector_float_sum() 42 …e void vector_float_sum(float32x4_t &result, float32x4_t &result_square, const float16x8_t &inputs) in vector_float_sum() 48 inline float16x8_t vector_float_norm(const float16x8_t &inputs, const float32x4_t &vec_mean, const … in vector_float_norm() 61 InputType vector_float_norm(const InputType &inputs, const AccType &vec_mean, const AccType &vec_mu… in vector_float_norm() 157 const auto val = static_cast<AccType>(*(input_ptr + x)); in instance_normalization_nchw() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | dataset.cc | 229 const std::vector<Node*>& inputs, in AddDataset() 235 const DatasetBase* dataset, const std::vector<Node*>& inputs, in AddDataset() 247 const std::vector<std::pair<size_t, Node*>>& inputs, in AddDataset() 257 const std::vector<std::pair<size_t, Node*>>& inputs, in AddDataset() 366 void GraphDefBuilderWrapper::AddPlaceholderInternal(const Tensor& val, in AddPlaceholderInternal() 373 void GraphDefBuilderWrapper::AddTensorInternal(const Tensor& val, in AddTensorInternal() 595 std::vector<const DatasetBase*> inputs; in ComputeNumSources() local 659 std::vector<const DatasetBase*> inputs; in MergeOptionsFromInputs() local 687 std::vector<const DatasetBase*> inputs; in MakeIterator() local 711 std::vector<const DatasetBase*> inputs; in MakeSplitProviders() local
|
/aosp_15_r20/external/libaom/av1/encoder/arm/ |
H A D | ml_neon.c | 32 static void nn_propagate_8to1(int num_inputs, const float *const inputs, in nn_propagate_8to1() 62 static void nn_propagate_xto1(int num_inputs, const float *const inputs, in nn_propagate_xto1() 97 static void nn_propagate_xsto1(int num_inputs, const float *const inputs, in nn_propagate_xsto1() 116 static void nn_propagate_4to1(int num_inputs, const float *const inputs, in nn_propagate_4to1() 142 static void nn_propagate_4to4(int num_inputs, const float *const inputs, in nn_propagate_4to4() 180 static void nn_propagate_4to8(const int num_inputs, const float *const inputs, in nn_propagate_4to8() 225 static void nn_propagate_8to4(const int num_inputs, const float *const inputs, in nn_propagate_8to4() 326 float val = layer_bias[node]; in av1_nn_predict_neon() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/parse_example/ |
H A D | parse_example_test.cc | 375 std::vector<std::vector<std::string>> inputs(num_tests); in TEST() local 378 float val = i; in TEST() local 409 std::vector<std::vector<std::string>> inputs(num_tests); in TEST() local 412 float val = i; in TEST() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ |
H A D | flatbuffer_export.cc | 478 std::map<std::string, std::string> inputs; member 736 std::string Translator::UniqueName(mlir::Value val) { in UniqueName() 1007 auto inputs = builder_.CreateVector(operands); in BuildIfOperator() local 1023 auto inputs = builder_.CreateVector(operands); in BuildCallOnceOperator() local 1049 auto inputs = builder_.CreateVector(operands); in BuildWhileOperator() local 1364 auto inputs = builder_.CreateVector(operands); in BuildOperator() local 1555 for (auto val : inst.getResults()) { in BuildSubGraph() local 1623 std::vector<int32_t> inputs, outputs; in BuildSubGraph() local 1683 const std::string& val = kv.second; in CreateMetadataVector() local 1841 auto inputs = GetList(subgraph_index, signature_def_data.inputs); in CreateSignatureDefs() local
|
/aosp_15_r20/frameworks/base/core/tests/coretests/src/com/android/internal/os/ |
H A D | BinderHeavyHitterTest.java | 83 int val = rand.nextInt(total - i); in generateRandomInput() local 136 ArrayList<HeavyHitterContainer> inputs = generateRandomInput(1000, hitters, counts); in testPositive() local 165 ArrayList<HeavyHitterContainer> inputs = generateRandomInput(1000, hitters, counts); in testNegative() local
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | logging_ops.py | 146 def print_v2(*inputs, **kwargs): 403 def _Collect(val, collections, default_collections): argument 573 def merge_summary(inputs, collections=None, name=None): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/xla/ |
H A D | xla.py | 67 def compile(computation, inputs=None): # pylint: disable=redefined-builtin argument 268 def AddValue(self, val): argument 306 def _compile_internal(computation, inputs=None): argument
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/operators/ |
H A D | misc.cpp | 87 void promoteInputs(std::vector<ExprHandle>& inputs, const int typeConstraints) { in promoteInputs() 164 auto val = intValue(dim.node()); in getTensorInfo() local 319 const std::vector<ArgValue>& inputs, in computeChunk() 353 const std::vector<ArgValue>& inputs, in computeTranspose() 380 const std::vector<ArgValue>& inputs, in computeExpand() 394 const std::vector<ArgValue>& inputs, in computeReshape() 461 const std::vector<ArgValue>& inputs, in computeFlatten() 509 const std::vector<ArgValue>& inputs, in computeCatWoConditionals() 618 const std::vector<ArgValue>& inputs, in computeCat() 680 const std::vector<ArgValue>& inputs, in computeEmbedding()
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/benchmark/ |
H A D | benchmark_model.cc | 76 const std::vector<InputLayerInfo>& inputs, in CreateTensorsFromInputInfo() 121 Status GetOutputShapes(const std::vector<InputLayerInfo>& inputs, in GetOutputShapes() 154 const std::vector<InputLayerInfo>& inputs, in CalculateFlops() 288 Status RunBenchmark(const std::vector<InputLayerInfo>& inputs, in RunBenchmark() 326 const std::vector<InputLayerInfo>& inputs, in TimeMultipleRuns() 535 std::vector<InputLayerInfo> inputs; in Main() local 562 float val; in Main() local
|
/aosp_15_r20/external/tensorflow/tensorflow/c/ |
H A D | kernels_test.cc | 344 std::unique_ptr<char[]> val(new char[5]); in TEST_F() local 423 TF_Tensor* val; in TEST_F() local 542 int64_t val; in TEST_F() local 592 float val; in TEST_F() local 642 unsigned char val; in TEST_F() local 692 TF_DataType val; in TEST_F() local 815 gtl::InlinedVector<TensorValue, 4> inputs; in TEST() local 1298 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_F() local 1372 gtl::InlinedVector<TensorValue, 4> inputs; in TEST_F() local
|
/aosp_15_r20/external/pytorch/torch/_dynamo/ |
H A D | debug_utils.py | 415 def cast_to(dtype, model, inputs): argument 433 def cast_to_fp64(model, inputs): argument 580 def symint(self, val): argument 684 def symint(self, name, val) -> None: argument
|
/aosp_15_r20/external/pytorch/torch/csrc/cuda/ |
H A D | nccl.cpp | 152 const char* val = getenv("TORCH_NCCL_NONBLOCKING_TIMEOUT"); in _parse_nccl_nonblocking_timeout() local 280 ArrayRef<ncclComm_t> get_communicators(TensorList inputs) { in get_communicators() 339 TensorList inputs, in check_inputs() 378 TensorList inputs, in check_inputs() 619 const std::vector<at::Tensor>& inputs, in reduce() 668 std::vector<at::Tensor>& inputs, in reduce() 677 const std::vector<at::Tensor>& inputs, in all_reduce() 719 const std::vector<at::Tensor>& inputs, in reduce_scatter() 761 const std::vector<at::Tensor>& inputs, in all_gather() 1030 const at::Tensor& inputs, in gather() [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
H A D | execute_node.h | 66 eager::RemoteTensorHandle* val) const override { in GetRemoteArg() 95 const absl::InlinedVector<TensorHandle*, 4>& inputs, in ExecuteNode() 151 const absl::InlinedVector<TensorHandle*, 4>& inputs, in AsyncExecuteNode()
|
/aosp_15_r20/external/pytorch/benchmarks/tensorexpr/ |
H A D | benchmark.py | 264 def inputs(self): member in DynamicShape 269 def inputs(self, val): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
H A D | custom_training_loop_input_test.py | 549 def computation(inputs): argument 597 def run(inputs): argument 650 def run(inputs): argument 758 def run(inputs): argument 783 def embedding_lookup(inputs): argument 973 def func(inputs): argument 1003 def train_step(val): argument
|
/aosp_15_r20/external/pytorch/torch/csrc/profiler/standalone/ |
H A D | execution_trace_observer.cpp | 67 inline std::string getScalarValue(const c10::IValue& val) { in getScalarValue() 232 const std::string& inputs = "[]", in writeJsonNode() 355 const c10::IValue& val, in convertIValue() 463 const c10::IValue& val, in appendValueInfo() 594 const auto inputs = fn.inputs(); in recordOperatorStart() local
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
H A D | impl.cpp | 59 std::string iValueToString(const c10::IValue& val) { in iValueToString() 67 const auto& inputs = node->inputs(); in allArgsAreTensors() local 147 for (const auto* val : value_set) { in dumpValueSet() local 442 auto inputs = collectValuesWithTrackedLifetimes(node->inputs()); in ManagedTensorRanges() local 1859 const Value* val = pnode.node()->output(i); in check_for_memory_leak() local 1939 const Value* val = pnode.node()->output(i); in checkOutputTensorMemoryLeaks() local 2049 ProcessedNodeInputs inputs, in StaticNodeInfo() 2087 const auto inputs = inputs_ivalue_vec(); in run() local
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
H A D | tensorexpr_fuser.cpp | 154 void setTensorExprFuserEnabled(bool val) { in setTensorExprFuserEnabled() 173 void setTensorExprDynamicShapeFusionEnabled(bool val) { in setTensorExprDynamicShapeFusionEnabled() 437 auto inputs = fusion_group->inputs(); in buildShapeExpressions() local 580 value_list sortReverseTopological(ArrayRef<Value*> inputs, Block* b) { in sortReverseTopological() 607 auto inputs = sortReverseTopological( in createFusionGroup() local
|