Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/linalg/
Dlinalg_ops_common.cc178 for (int input_idx = 0; input_idx < context->num_inputs(); ++input_idx) { in PrepareOutputs() local
202 for (int input_idx : unused_inputs) { in PrepareOutputs() local
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model.cc106 const int32_t input_idx = input.first; in PopulateRealValueOpSet() local
224 int32_t input_idx) { in InputQuantizeRequired()
405 const int32_t input_idx = SetInputType( in SetInputAndOutputTypes() local
472 for (size_t input_idx = 0; input_idx < op->inputs.size(); ++input_idx) { in ApplyConstraints() local
586 for (const int32_t input_idx : subgraph->inputs) { in IsSubgraphInput() local
600 int32_t input_idx = input.first; in QuantizeOpInput() local
1191 const int32_t input_idx = input.first; in FillQuantizationParams() local
Dmodify_model_interface.cc57 for (size_t input_idx = 0; input_idx < subgraph->inputs.size(); in GetInputTensors() local
400 for (size_t input_idx = 0; input_idx < subgraph->inputs.size(); in AddUint8Dequant() local
Dquantize_model_test.cc283 const int32_t input_idx = subgraph->inputs[0]; in TEST_P() local
350 const int32_t input_idx = subgraph->inputs[0]; in TEST_P() local
1021 for (size_t input_idx = 0; input_idx < 2; ++input_idx) { in TEST_F() local
1089 for (size_t input_idx = 0; input_idx < 2; ++input_idx) { in TEST_P() local
Dquantize_weights.cc400 const auto input_idx = GetWeightInputIndices(op_code, custom_op_map); in PassQuantizationAndGetConsumers() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dkernel_thunk.cc72 int input_idx = 0; in PrintBufferContents() local
/external/tensorflow/tensorflow/core/grappler/
Dgraph_topology_view.cc125 const int input_idx = it->second; in InitializeFromGraph() local
/external/tensorflow/tensorflow/core/kernels/
Dstateful_random_ops.cc155 Status GetScalar(const Tensor& tensor, int input_idx, T* result) { in GetScalar()
172 Status GetAlg(OpKernelContext* ctx, int input_idx, Algorithm* alg) { in GetAlg()
Daggregate_ops.cc57 for (int input_idx = 0; input_idx < num; ++input_idx) { in Compute() local
Dstateless_random_ops_v2.cc51 Status GetScalar(const Tensor& tensor, int input_idx, T* result) { in GetScalar()
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration.cc116 int input_idx = sparse_keys_start + i; in ExtractExampleParserConfiguration() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc97 for (int input_idx = 0; input_idx < node->inputs->size; ++input_idx) { in Eval() local
/external/mesa3d/src/broadcom/compiler/
Dnir_to_vir.c691 int8_t input_idx, uint8_t swizzle, int array_index) in emit_fragment_varying()
793 int input_idx = (base_attr + array_index) * 4 + chan; in emit_fragment_input() local
807 int input_idx = (attr + loc_offset) * 4 + chan; in emit_compact_fragment_input() local
1745 unsigned input_idx = c->num_inputs++; in ntq_setup_gs_inputs() local
2729 const uint32_t input_idx = in ntq_emit_intrinsic() local
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_inspection_required_ops_utils.cc168 Status AddInputIdentity(Node* node, int input_idx, Graph* graph, in AddInputIdentity()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dconv.cc91 int input_idx = in ConvPerChannel() local
Ddepthwise_conv.cc128 int input_idx = in DepthwiseConvPerChannel() local
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.cc211 const int input_idx = interpreter_->inputs()[i]; in BuildInterpreter() local
/external/tensorflow/tensorflow/compiler/jit/
Dnode_matchers.cc172 for (int input_idx = 0, e = input_matchers->size(); input_idx < e; in MatchAndExplain() local
Dbuild_xla_ops_pass.cc387 int input_idx = 0; in GetXlaRunArgs() local
Dxla_launch_util.cc126 for (int input_idx = 0; input_idx < ctx->num_inputs(); input_idx++) { in InputsFromContext() local
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc718 int input_idx, ShapeHandle* out) { in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
736 Status InferenceContext::MakeShapeFromShapeTensor(int input_idx, in MakeShapeFromShapeTensor()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_grad_filter_ops.cc633 size_t input_idx = 0; in MakeInputTfShape() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfusion_utils.cc244 auto input_idx = std::distance(second_inputs.begin(), input_it); in FuseReturns() local
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dnativeinterpreterwrapper_jni.cc629 jint input_idx, jintArray dims, jboolean strict) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_resizeInput()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.h345 void AddInput(const string& input, int64 output_index, int input_idx) { in AddInput()

12