| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_slice_grad_op.cc | 30 const Tensor *backprop_val_grad, *input_indices, *output_indices, *input_start; in Compute() local
32 OP_REQUIRES_OK(ctx, ctx->input("input_indices", &input_indices)); in Compute()
37 TensorShapeUtils::IsMatrix(input_indices->shape()) && in Compute()
42 input_indices->shape().DebugString(), " and ", in Compute()
51 input_indices->dim_size(1) == output_indices->dim_size(1), in Compute()
53 "ndims: got: ", input_indices->dim_size(1), " and ", in Compute()
56 ctx, output_indices->dim_size(0) <= input_indices->dim_size(0), in Compute()
60 input_indices->dim_size(0))); in Compute()
72 const int num_dims = input_indices->dim_size(1); in Compute()
78 const int64 input_nnz = input_indices->dim_size(0); in Compute()
[all …]
|
| D | dynamic_stitch_op_gpu.cu.cc | 34 GpuDeviceArrayStruct<int32> input_indices, in DynamicStitchKernel() argument
37 int32* data_indices = GetGpuDeviceArrayOnDevice(&input_indices); in DynamicStitchKernel()
54 const GpuDeviceArrayStruct<int>& input_indices, in DynamicStitchGPUImpl() argument
62 slice_size, output_size, input_indices, in DynamicStitchGPUImpl()
70 const GpuDeviceArrayStruct<int32>& input_indices, \
|
| D | sparse_split_op.cc | 34 const Tensor& input_indices = context->input(1); in Compute() local
38 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()), in Compute()
41 input_indices.shape().DebugString())); in Compute()
45 input_indices.shape().DebugString())); in Compute()
69 input_indices, input_values, in Compute()
|
| D | sparse_slice_op.cc | 31 const Tensor& input_indices = context->input(0); in Compute() local
37 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices.shape()), in Compute()
40 input_indices.shape().DebugString())); in Compute()
44 input_indices.shape().DebugString())); in Compute()
72 input_indices, input_values, in Compute()
|
| D | serialize_sparse_op.cc | 57 const Tensor* input_indices; in Compute() local
61 OP_REQUIRES_OK(context, context->input("sparse_indices", &input_indices)); in Compute()
64 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices->shape()), in Compute()
67 input_indices->shape().DebugString())); in Compute()
83 OP_REQUIRES_OK(context, Serialize(*input_indices, &serialized_sparse_t(0))); in Compute()
339 const Tensor* input_indices; in Compute() local
342 OP_REQUIRES_OK(context, context->input("sparse_indices", &input_indices)); in Compute()
345 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices->shape()), in Compute()
348 input_indices->shape().DebugString())); in Compute()
371 OP_REQUIRES_OK(context, SparseTensor::Create(*input_indices, *input_values, in Compute()
|
| D | aggregate_ops.cc | 54 gtl::InlinedVector<int, 8> input_indices(num); in Compute() local
55 std::iota(input_indices.begin(), input_indices.end(), 0); in Compute()
69 input_indices[0] = reused_input; in Compute()
70 input_indices[reused_input] = 0; in Compute()
74 #define I(IDX) ctx->input(input_indices[IDX]).template flat<T>() in Compute()
|
| D | sparse_tensors_map_ops.cc | 170 const Tensor* input_indices; in Compute() local
175 OP_REQUIRES_OK(context, context->input("sparse_indices", &input_indices)); in Compute()
180 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices->shape()), in Compute()
183 input_indices->shape().DebugString())); in Compute()
201 OP_REQUIRES_OK(context, SparseTensor::Create(*input_indices, *input_values, in Compute()
225 const Tensor* input_indices; in Compute() local
230 OP_REQUIRES_OK(context, context->input("sparse_indices", &input_indices)); in Compute()
235 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices->shape()), in Compute()
238 input_indices->shape().DebugString())); in Compute()
261 OP_REQUIRES_OK(context, SparseTensor::Create(*input_indices, *input_values, in Compute()
|
| D | reshape_util.cc | 42 typename TTypes<int64>::ConstMatrix input_indices, in operator ()()
46 const int64 nnz = input_indices.dimension(0); in operator ()()
67 id += input_indices(i, j) * input_strides[j]; in operator ()()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseReorder.pbtxt | 4 name: "input_indices"
13 1-D. `N` non-empty values corresponding to `input_indices`.
25 2-D. `N x R` matrix with the same indices as input_indices, but
43 If the tensor has rank `R` and `N` non-empty values, `input_indices` has
|
| D | api_def_SparseReshape.pbtxt | 4 name: "input_indices"
40 tensor. The `input_indices` are recomputed based on the requested `new_shape`.
51 has length `R_out`, then `input_indices` has shape `[N, R_in]`,
|
| D | api_def_SparseReduceSumSparse.pbtxt | 4 name: "input_indices"
13 1-D. `N` non-empty values corresponding to `input_indices`.
|
| D | api_def_SparseReduceMaxSparse.pbtxt | 4 name: "input_indices"
13 1-D. `N` non-empty values corresponding to `input_indices`.
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | quantization_util.cc | 90 TfLiteContext* context, const std::vector<uint32_t>& input_indices, in DequantizeInputs() argument
92 for (auto index : input_indices) { in DequantizeInputs()
99 TfLiteContext* context, const std::vector<int64_t>& input_indices, in DequantizeInputs() argument
101 for (auto index : input_indices) { in DequantizeInputs()
|
| D | quantization_util.h | 36 TfLiteContext* context, const std::vector<uint32_t>& input_indices,
40 TfLiteContext* context, const std::vector<int64_t>& input_indices,
|
| D | quantization_util_test.cc | 163 std::vector<uint32_t> input_indices = {1}; in TEST() local
166 auto status = DequantizeInputs(&context, input_indices, quant_conversion_map); in TEST()
188 std::vector<int64_t> input_indices = {1}; in TEST() local
191 auto status = DequantizeInputs(&context, input_indices, quant_conversion_map); in TEST()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SparseReduceMax.pbtxt | 4 name: "input_indices"
51 name: "input_indices"
100 name: "input_indices"
150 name: "input_indices"
|
| D | SparseReduceSum.pbtxt | 4 name: "input_indices"
56 name: "input_indices"
110 name: "input_indices"
165 name: "input_indices"
|
| D | SparseReduceMaxSparse.pbtxt | 4 name: "input_indices"
59 name: "input_indices"
116 name: "input_indices"
174 name: "input_indices"
|
| D | SparseReduceSumSparse.pbtxt | 4 name: "input_indices"
64 name: "input_indices"
126 name: "input_indices"
189 name: "input_indices"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SparseReduceMax.pbtxt | 4 name: "input_indices"
51 name: "input_indices"
100 name: "input_indices"
150 name: "input_indices"
|
| D | SparseReduceMaxSparse.pbtxt | 4 name: "input_indices"
59 name: "input_indices"
116 name: "input_indices"
174 name: "input_indices"
|
| D | SparseReduceSum.pbtxt | 4 name: "input_indices"
56 name: "input_indices"
110 name: "input_indices"
165 name: "input_indices"
|
| D | SparseReduceSumSparse.pbtxt | 4 name: "input_indices"
64 name: "input_indices"
126 name: "input_indices"
189 name: "input_indices"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_grad.py | 50 input_indices = op.inputs[0]
53 num_entries = array_ops.shape(input_indices)[0]
56 input_indices, entry_indices, input_shape)
137 input_indices = op.inputs[0]
142 backprop_val_grad, input_indices, input_start, output_indices)
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_jni.cc | 214 jintArray input_indices = env->NewIntArray(ninputs); in buildSubgraph() local
220 jint* input_indices_elems = env->GetIntArrayElements(input_indices, nullptr); in buildSubgraph()
237 env->ReleaseIntArrayElements(input_indices, input_indices_elems, 0); in buildSubgraph()
245 input_handles, input_indices, output_handles, output_indices); in buildSubgraph()
258 jintArray input_indices, jstring name, jobject cond_graph_builder, in Java_org_tensorflow_Graph_whileLoop() argument
267 resolveOutputs(env, "inputs", input_handles, input_indices, inputs.get(), in Java_org_tensorflow_Graph_whileLoop()
|