Home
last modified time | relevance | path

Searched defs:input_idx (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dlinalg_ops_common.cc176 for (int input_idx = 0; input_idx < context->num_inputs(); ++input_idx) { in PrepareOutputs() local
200 for (int input_idx : unused_inputs) { in PrepareOutputs() local
Daggregate_ops.cc62 for (int input_idx = 0; input_idx < num; ++input_idx) { in Compute() local
Dmkl_conv_grad_filter_ops.cc622 size_t input_idx = 0; in MakeInputTfShape() local
Deigen_spatial_convolutions_test.cc1523 int input_idx = in PackRhsHelper() local
/external/tensorflow/tensorflow/core/grappler/
Dgraph_topology_view.cc125 const int input_idx = it->second; in InitializeFromGraph() local
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.cc128 const int input_idx = interpreter_->inputs()[i]; in BuildInterpreter() local
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration.cc116 int input_idx = sparse_keys_start + i; in ExtractExampleParserConfiguration() local
/external/mesa3d/src/gallium/drivers/radeonsi/
Dsi_shader_nir.c654 unsigned input_idx = variable->data.driver_location; in si_nir_build_llvm() local
/external/tensorflow/tensorflow/lite/tools/optimize/
Dsubgraph_quantizer.cc335 for (size_t input_idx = 0; input_idx < op->inputs.size(); ++input_idx) { in AsymmetricQuantizeInputsAndOutputs() local
Dsubgraph_quantizer_test.cc477 for (size_t input_idx = 0; input_idx < 2; ++input_idx) { in TEST() local
/external/tensorflow/tensorflow/compiler/jit/
Dnode_matchers.cc168 for (int input_idx = 0, e = input_matchers->size(); input_idx < e; in MatchAndExplain() local
/external/u-boot/drivers/net/
Dsmc91111.c890 int input_idx; in smc_read_phy_register() local
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc776 int input_idx, ShapeHandle* out) { in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
793 Status InferenceContext::MakeShapeFromShapeTensor(int input_idx, in MakeShapeFromShapeTensor()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfusion_utils.cc243 auto input_idx = std::distance(second_inputs.begin(), input_it); in FuseReturns() local
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dnativeinterpreterwrapper_jni.cc432 jint input_idx, jintArray dims) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_resizeInput()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc1509 const int input_idx = input_placeholder_idx->second; in PlaceInlinedFunctionBody() local
1769 const int input_idx = input_placeholder_idx->second; in InlineIndirectFunctionCall() local
Dconstant_folding.cc3018 for (int input_idx = 0; input_idx < node->input_size(); ++input_idx) { in PartialConstPropThroughIdentityN() local
/external/v8/src/wasm/baseline/
Dliftoff-compiler.cc228 uint32_t input_idx = input_idx_start; in CollectReservedRegsForParameters() local
253 uint32_t ProcessParameter(ValueType type, uint32_t input_idx) { in ProcessParameter()
341 uint32_t input_idx = kInstanceParameterIndex + 1; in StartFunctionBody() local
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.h346 void AddInput(const string& input, int64 output_index, int input_idx) { in AddInput()
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc1580 const gtl::InlinedVector<std::pair<Node*, int>, 4>& inputs, int* input_idx, in GetNodesProducingTFTensorList() argument
1645 const gtl::InlinedVector<std::pair<Node*, int>, 4>& inputs, int* input_idx, in GetNodesProducingMklTensorList() argument
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h944 inline TensorShape GetTfShape(OpKernelContext* context, size_t input_idx) { in GetTfShape()
/external/virglrenderer/src/
Dvrend_shader.c2288 int input_idx, in create_swizzled_clipdist()
/external/tensorflow/tensorflow/python/framework/
Dops.py2205 def _tf_input(self, input_idx): argument
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc4276 int input_idx = 0; in ConvertSegmentToGraphDef() local
/external/vulkan-validation-layers/layers/
Dcore_validation.cpp6883 for (uint32_t input_idx = 0; input_idx < set_count; input_idx++) { in UpdateLastBoundDescriptorSets() local