/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 34 OP_REQUIRES_OK(ctx, ctx->input("output_indices", &output_indices)); in Compute() 38 TensorShapeUtils::IsMatrix(output_indices->shape()), in Compute() 43 output_indices->shape().DebugString())); in Compute() 51 input_indices->dim_size(1) == output_indices->dim_size(1), in Compute() 54 output_indices->dim_size(1))); in Compute() 56 ctx, output_indices->dim_size(0) <= input_indices->dim_size(0), in Compute() 59 output_indices->dim_size(0), " and ", in Compute() 62 ctx, backprop_val_grad->NumElements() == output_indices->dim_size(0), in Compute() 66 output_indices->dim_size(0))); in Compute() [all …]
|
D | deserialize_sparse_variant_op.cc | 59 const Tensor* output_indices; in Compute() local 69 output_shape->NumElements(), &output_indices, in Compute() 71 context->set_output(0, *output_indices); in Compute() 152 Tensor* output_indices; in Compute() local 159 &output_indices)); in Compute() 169 int64* output_indices_data = output_indices->matrix<int64>().data(); in Compute() 304 int index, int expected_rank, const Tensor** output_indices, in GetAndValidateSparseTensorIndicesAndValues() argument 307 *output_indices = serialized_indices.get<Tensor>(); in GetAndValidateSparseTensorIndicesAndValues() 308 if (*output_indices == nullptr) { in GetAndValidateSparseTensorIndicesAndValues() 312 if ((*output_indices)->dtype() != DT_INT64) { in GetAndValidateSparseTensorIndicesAndValues() [all …]
|
D | deserialize_sparse_string_op.cc | 82 Tensor output_indices; in Compute() local 89 &output_indices, &output_values, &output_shape)); in Compute() 90 context->set_output(0, output_indices); in Compute() 105 Tensor output_indices; in Compute() local 111 serialized_sparse_t(i, 2), dtype_, i, &output_indices, in Compute() 113 int64 num_entries = output_indices.dim_size(0); in Compute() 114 int rank = output_indices.dim_size(1); in Compute() 119 const auto& output_indices_t = output_indices.matrix<int64>(); in Compute() 232 Tensor* output_indices, Tensor* output_values, Tensor* output_shape) { in GetAndValidateSparseTensor() argument 234 TF_RETURN_IF_ERROR(this->Deserialize(serialized_indices, output_indices)); in GetAndValidateSparseTensor() [all …]
|
D | sparse_tensors_map_ops.cc | 296 Tensor output_indices = Tensor(DT_INT64, {num_entries, rank - 1}); in Compute() local 299 auto output_indices_t = output_indices.matrix<int64>(); in Compute() 311 SparseTensor::Create(output_indices, output_values, in Compute() 388 const Tensor& output_indices = st.indices(); in Compute() local 392 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(output_indices.shape()), in Compute() 396 output_indices.shape().DebugString())); in Compute() 409 int64 num_entries = output_indices.dim_size(0); in Compute() 416 int rank = output_indices.dim_size(1); in Compute() 428 DT_INT64, TensorShape({num_entries, 1 + output_indices.dim_size(1)})); in Compute() 430 const auto& output_indices_t = output_indices.matrix<int64>(); in Compute()
|
D | serialize_sparse_op.cc | 193 Tensor output_indices = Tensor(DT_INT64, {num_entries, rank - 1}); in operator ()() local 196 auto output_indices_t = output_indices.matrix<int64>(); in operator ()() 206 serialize(output_indices, &serialized_sparse_t(b, 0)); in operator ()() 292 Tensor& output_indices = serialized_sparse_t(b, 0).emplace<Tensor>( in operator ()() local 298 static_cast<int64*>(DMAHelper::base(&output_indices)); in operator ()()
|
D | sparse_fill_empty_rows_op.cc | 143 auto output_indices = output_indices_t->matrix<Tindex>(); in operator ()() local 158 std::copy_n(&indices(i, 0), rank, &output_indices(output_i, 0)); in operator ()() 173 output_indices(starting_index, 0) = row; in operator ()() 175 output_indices(starting_index, col) = 0; in operator ()()
|
/external/libtextclassifier/native/utils/tflite/ |
D | dist_diversification.cc | 83 TfLiteTensor& output_indices = in AllocateOutputIndexes() local 86 return context->ResizeTensor(context, &output_indices, in AllocateOutputIndexes() 97 TfLiteTensor& output_indices = in Prepare() local 100 tflite::SetTensorToDynamic(&output_indices); in Prepare() 110 TfLiteTensor& output_indices = in Eval() local 113 if (tflite::IsDynamicTensor(&output_indices)) { in Eval() 133 std::copy(indices.begin(), indices.end(), output_indices.data.i32); in Eval() 134 std::fill_n(output_indices.data.i32 + indices.size(), in Eval()
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | reverse_test.cc | 94 std::vector<int64> output_indices(spec.input_dims.size()); in TEST_P() local 97 output_indices[i] = indices[i]; in TEST_P() 101 output_indices[dim] = (spec.input_dims[dim] - 1) - indices[dim]; in TEST_P() 103 expected.Set<float>(output_indices, value); in TEST_P()
|
/external/tensorflow/tensorflow/core/kernels/image/ |
D | non_max_suppression_op.cu.cc | 421 Tensor* output_indices = nullptr; in DoNMS() local 423 context->allocate_output(0, TensorShape({0}), &output_indices)); in DoNMS() 494 Tensor* output_indices = nullptr; in DoNMS() local 500 &output_indices)); in DoNMS() 519 Tensor* output_indices = nullptr; in DoNMS() local 523 0, TensorShape({max_output_size}), &output_indices)); in DoNMS() 528 output_indices->flat<int>().data())); in DoNMS() 532 &output_indices)); in DoNMS() 543 (*output_indices).flat<int>().data())); in DoNMS() 617 Tensor* output_indices = nullptr; in Compute() local [all …]
|
/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
D | quantization_util.cc | 108 TfLiteContext* context, const std::vector<uint32_t>& output_indices, in QuantizeOutputs() argument 110 for (auto index : output_indices) { in QuantizeOutputs() 118 TfLiteContext* context, const std::vector<int64_t>& output_indices, in QuantizeOutputs() argument 120 for (auto index : output_indices) { in QuantizeOutputs()
|
D | quantization_util.h | 49 TfLiteContext* context, const std::vector<uint32_t>& output_indices, 53 TfLiteContext* context, const std::vector<int64_t>& output_indices,
|
D | quantization_util_test.cc | 211 std::vector<uint32_t> output_indices = {0}; in TEST() local 214 auto status = QuantizeOutputs(&context, output_indices, quant_conversion_map); in TEST() 233 std::vector<int64_t> output_indices = {0}; in TEST() local 236 auto status = QuantizeOutputs(&context, output_indices, quant_conversion_map); in TEST()
|
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
D | pooling.cc | 146 bool stride_correction, bool output_indices, in GetMaxPoolingKernelCode() argument 158 if (output_indices) { in GetMaxPoolingKernelCode() 211 if (output_indices) { in GetMaxPoolingKernelCode() 245 if (output_indices) { in GetMaxPoolingKernelCode() 279 if (output_indices) { in GetMaxPoolingKernelCode() 303 attr.output_indices, &op); in CreatePooling() 327 attr.output_indices, &op); in CreatePooling()
|
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
D | SparseSparseMaximum.pbtxt | 28 name: "output_indices" 80 name: "output_indices" 134 name: "output_indices" 189 name: "output_indices"
|
D | SparseReduceMaxSparse.pbtxt | 20 name: "output_indices" 75 name: "output_indices" 132 name: "output_indices" 190 name: "output_indices"
|
D | SparseReduceSumSparse.pbtxt | 20 name: "output_indices" 80 name: "output_indices" 142 name: "output_indices" 205 name: "output_indices"
|
D | SparseSparseMinimum.pbtxt | 28 name: "output_indices" 85 name: "output_indices" 144 name: "output_indices" 204 name: "output_indices"
|
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
D | SparseReduceMaxSparse.pbtxt | 20 name: "output_indices" 75 name: "output_indices" 132 name: "output_indices" 190 name: "output_indices"
|
D | SparseSparseMaximum.pbtxt | 28 name: "output_indices" 80 name: "output_indices" 134 name: "output_indices" 189 name: "output_indices"
|
D | SparseReduceSumSparse.pbtxt | 20 name: "output_indices" 80 name: "output_indices" 142 name: "output_indices" 205 name: "output_indices"
|
D | SparseSparseMinimum.pbtxt | 28 name: "output_indices" 85 name: "output_indices" 144 name: "output_indices" 204 name: "output_indices"
|
/external/tensorflow/tensorflow/core/util/sparse/ |
D | sparse_tensor.h | 498 std::vector<Tensor> output_indices; in Split() local 501 output_indices.reserve(num_split); in Split() 535 output_indices.emplace_back(DT_INT64, in Split() 540 output_indices_t.emplace_back(output_indices[i].matrix<int64>()); in Split() 566 Create(output_indices[i], output_values[i], output_shapes[i], &tensor); in Split() 629 Tensor output_indices(DT_INT64, TensorShape({count, dims})); in Slice() 632 auto output_indices_t = output_indices.matrix<int64>(); in Slice() 659 return SparseTensor(output_indices, output_values, output_shape); in Slice()
|
/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | resolve_constant_binary.cc | 127 const std::vector<int> output_indices = ReverseOffset(output_shape, k); in EvaluateBinaryOperatorOnConstantInputs() local 129 modulo_indices[i] = output_indices[i] % small_sizes[i]; in EvaluateBinaryOperatorOnConstantInputs() 131 PairwiseVectorSelect(input0_larger, output_indices, modulo_indices, in EvaluateBinaryOperatorOnConstantInputs() 164 output_data[Offset(output_shape, output_indices)] = outval; in EvaluateBinaryOperatorOnConstantInputs()
|
/external/tensorflow/tensorflow/core/api_def/base_api/ |
D | api_def_SparseReorder.pbtxt | 23 name: "output_indices" 32 1-D. `N` non-empty values corresponding to `output_indices`.
|
/external/tensorflow/tensorflow/python/ops/ |
D | sparse_ops.py | 1164 output_indices, output_values, output_shape = gen_sparse_ops.sparse_slice( 1172 return sparse_tensor.SparseTensor(output_indices, output_values, 2153 (output_indices, output_values, empty_row_indicator, 2160 indices=output_indices, 2335 output_indices, output_values, output_shape = ( 2339 output_indices.set_shape([None, rank]) 2342 return sparse_tensor.SparseTensor(output_indices, output_values, output_shape) 2408 output_indices, output_values, output_shape = ( 2413 output_indices.set_shape([None, rank]) 2416 return sparse_tensor.SparseTensor(output_indices, output_values, output_shape) [all …]
|