Searched defs:input_indices (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_stitch_op_gpu.cu.cc | 34 GpuDeviceArrayStruct<int32> input_indices, in DynamicStitchKernel()
54 const GpuDeviceArrayStruct<int>& input_indices, in DynamicStitchGPUImpl()
|
| D | sparse_split_op.cc | 34 const Tensor& input_indices = context->input(1); in Compute() local
|
| D | sparse_slice_op.cc | 31 const Tensor& input_indices = context->input(0); in Compute() local
|
| D | sparse_slice_grad_op.cc | 30 const Tensor *backprop_val_grad, *input_indices, *output_indices, *input_start; in Compute() local
|
| D | serialize_sparse_op.cc | 55 const Tensor* input_indices; in Compute() local
162 const Tensor* input_indices; in Compute() local
|
| D | aggregate_ops.cc | 59 gtl::InlinedVector<int, 8> input_indices(num); in Compute() local
|
| D | sparse_tensors_map_ops.cc | 169 const Tensor* input_indices; in Compute() local
224 const Tensor* input_indices; in Compute() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | reorder_reshape_transpose.cc | 62 std::vector<int> input_indices; in ComputeNewPerm() local
|
| D | resolve_constant_unary.cc | 66 std::vector<int> input_indices = ReverseOffset(input_shape, input_offset); in ReduceGeneric() local
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_jni.cc | 214 jintArray input_indices = env->NewIntArray(ninputs); in buildSubgraph() local
258 jintArray input_indices, jstring name, jobject cond_graph_builder, in Java_org_tensorflow_Graph_whileLoop()
|
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| D | wals_solver_ops.cc | 77 const Tensor& input_indices = context->input(4); in Compute() local
|
| /external/libtextclassifier/annotator/ |
| D | annotator_jni.cc | 530 CodepointSpan input_indices = in TC3_JNI_METHOD() local
553 const CodepointSpan input_indices = in TC3_JNI_METHOD() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 569 ShapeHandle input_indices = c->input(0); in __anona7ffb82a1602() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.h | 1158 absl::Span<const int64> input_indices = output_indices; in Replicate() local
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen.cc | 185 const std::vector<int>* input_indices, in FlattenInputs()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1397 def _inputs_with_flattening(pfor_input, input_indices): argument
|