Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/contrib/lite/testing/
Dparse_testdata.cc134 for (int input_index = 0; input_index < inputs; input_index++) { in ParseExamples() local
142 for (int input_index = 0; input_index < outputs; input_index++) { in ParseExamples() local
155 int input_index = interpreter->inputs()[i]; in FeedExample() local
163 int input_index = interpreter->inputs()[i]; in FeedExample() local
/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.h189 const Tensor& GetInput(int input_index) const { in GetInput()
195 TensorValue mutable_input(int input_index) { in mutable_input()
Ddynamic_stitch_op_gpu.cu.cc42 const int32 input_index = data_indices[slice_id]; in DynamicStitchKernel() local
Dquantized_batch_norm_op.cc55 const int input_index = (row_index * depth) + channel; in ReferenceBatchNorm() local
148 const int input_index = (row_index * depth) + channel; in FixedPointBatchNorm() local
Dconv_ops_gpu_3.cu.cc193 int input_index = TensorIndexToFlat(input_tensor_index, input_dims); in SwapDimension0And2InTensor3Simple() local
219 int input_index = TensorIndexToFlat(input_tensor_index, input_dims); in SwapDimension1And2InTensor3Simple() local
305 int input_index = input_origin_flat_index + ti * input_dims[2] + tj; in SwapDimension1And2InTensor3UsingTiles() local
394 const int input_index = TensorIndexToFlat(input_tensor_index, input_dims); in PadInputCustomKernelNHWC() local
423 const int input_index = TensorIndexToFlat(input_tensor_index, input_dims); in PadInputCustomKernelNCHW() local
Ddeep_conv2d_test.cc30 const int input_index = k * cols + l; in ComputeKroneckerProduct() local
Dshape_ops.h32 inline Status GetRegularOrVariantShape(OpKernelContext* ctx, int input_index, in GetRegularOrVariantShape()
Dsegment_reduction_ops_gpu.cu.cc112 for (int input_index : CudaGridRangeX(input_total_size)) { in UnsortedSegmentCustomKernel() local
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dscatter_add_ndim_op.cc99 const int32 input_index = start_index + offset; in Compute() local
Dtree_utils.h212 int32 GetNumSparseFeatures(const T1& indices, int32 input_index, in GetNumSparseFeatures()
/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops.py285 def sort_key(input_index, a): argument
/external/v8/src/compiler/
Dnode.h188 int input_index() const { return InputIndexField::decode(bit_field_); } in input_index() function
244 Node* const* GetInputPtrConst(int input_index) const { in GetInputPtrConst()
248 Node** GetInputPtr(int input_index) { in GetInputPtr()
252 Use* GetUsePtr(int input_index) { in GetUsePtr()
Dint64-lowering.h65 int input_index; member
Dsimd-scalar-lowering.h66 int input_index; member
Dgraph-reducer.h143 int input_index; in NON_EXPORTED_BASE() member
/external/mesa3d/src/mesa/drivers/dri/i965/
Dgen8_sf_state.c96 int input_index = 0; in upload_sbe() local
Dgen6_sf_state.c210 int input_index = wm_prog_data->urb_setup[attr]; in calculate_attr_overrides() local
/external/tensorflow/tensorflow/contrib/tensorrt/
Dtensorrt_test.cc104 const int input_index = engine.getBindingIndex(kInputTensor); in Execute() local
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
Dquantize.cc171 std::size_t input_index, ArrayDataType* quantized_data_type, in ChooseQuantizationForOperatorInput()
439 for (std::size_t input_index = 0; input_index < op.inputs.size(); in Run() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc378 IrArray::Index input_index(index.size()); in MakeElementGenerator() local
427 IrArray::Index input_index = loops.AddLoopsForShapeOnDimensions( in MakeElementGenerator() local
/external/tensorflow/tensorflow/contrib/lite/toco/tflite/
Dimport.cc122 auto input_index = inputs->Get(i); in ImportOperators() local
/external/tensorflow/tensorflow/contrib/factorization/kernels/
Dwals_solver_ops.cc215 const int64 input_index = get_input_index(perm[shard.first]); in Compute() local
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc402 void OpKernelContext::forward_ref_input_to_ref_output(int input_index, in forward_ref_input_to_ref_output()
412 int input_index, int output_index, const TensorShape& output_shape, in forward_input_to_output_with_shape()
433 int input_index, output_index, stop; in forward_input_to_output_with_shape() local
459 int input_index, DataType output_dtype, const TensorShape& output_shape, in forward_input()
501 for (int input_index : candidate_input_indices) { in forward_input_or_allocate_temp() local
/external/v8/src/
Ddeoptimizer.cc805 int input_index = 0; in DoComputeJSFrame() local
1048 int input_index = 0; in DoComputeInterpretedFrame() local
1317 int input_index = 0; in DoComputeArgumentsAdaptorFrame() local
1506 int input_index = 0; in DoComputeConstructStubFrame() local
1696 int input_index = 0; in DoComputeAccessorStubFrame() local
1911 int input_index = 0; in DoComputeCompiledStubFrame() local
2134 TranslatedFrame::iterator* iterator, int* input_index, int frame_index, in WriteTranslatedValueToOutput()
2158 void Deoptimizer::WriteValueToOutput(Object* value, int input_index, in WriteValueToOutput()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.h186 int input_index; member

123