| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_split_op_gpu.cu.cc | 110 const Index* __restrict__ input_indices, int* __restrict__ slice_indexes) { in SparseSplitSliceIndexesKernel()
122 const Index* input_indices, in LaunchSparseSplitSliceIndexesKernel()
165 const Index* __restrict__ input_indices, const T* __restrict__ input_values, in SparseSplitScatterKernel()
194 const Index* slice_ends, const Index* input_indices, const T* input_values, in LaunchSparseSplitScatterKernel()
|
| H A D | dynamic_stitch_op_gpu.cu.cc | 34 GpuDeviceArrayStruct<int32> input_indices, in DynamicStitchKernel()
54 const GpuDeviceArrayStruct<int>& input_indices, in DynamicStitchGPUImpl()
|
| H A D | reshape_util_gpu.cu.cc | 34 const Tindex* __restrict__ input_indices, in ReshapeSparseTensorKernel()
61 typename TTypes<int64_t>::ConstMatrix input_indices, in operator ()()
|
| H A D | serialize_sparse_op.cc | 59 const Tensor* input_indices; in Compute() local
343 const Tensor* input_indices; in Compute() local
|
| H A D | sparse_tensors_map_ops.cc | 165 const Tensor* input_indices; in Compute() local
220 const Tensor* input_indices; in Compute() local
|
| H A D | sparse_slice_grad_op.cc | 80 const Tensor *backprop_val_grad, *input_indices, *output_indices, *input_start; in Compute() local
|
| H A D | sparse_slice_op.cc | 94 const Tensor& input_indices = context->input(0); in SparseSliceOpImpl() local
|
| H A D | sparse_split_op.cc | 78 const Tensor& input_indices = context->input(1); in SparseSplitOpImpl() local
|
| H A D | sparse_slice_grad_op_gpu.cu.cc | 96 MultiIndexSearchIterator input_indices, in SparseSliceGradKernel()
|
| H A D | aggregate_ops.cc | 55 gtl::InlinedVector<int, 8> input_indices(num); in Compute() local
|
| H A D | sparse_slice_op_gpu.cu.cc | 91 const int64_t* __restrict__ input_indices, in SparseSliceGatherKernel()
|
| H A D | topk_op_gpu.h | 453 Tensor input_indices; in LaunchSortKernel() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_reachability.cc | 53 absl::Span<const Index> input_indices, Index index) { in FastSetReachabilityToUnion()
59 absl::InlinedVector<Index, 16> input_indices; in SetReachabilityToUnionHelper() local
68 absl::Span<const Index> input_indices, Index index) { in SetReachabilityToUnionHelper()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | quantization_util.cc | 90 TfLiteContext* context, const std::vector<uint32_t>& input_indices, in DequantizeInputs()
99 TfLiteContext* context, const std::vector<int64_t>& input_indices, in DequantizeInputs()
|
| H A D | quantization_util_test.cc | 163 std::vector<uint32_t> input_indices = {1}; in TEST() local
188 std::vector<int64_t> input_indices = {1}; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/java/src/main/native/ |
| H A 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()
|
| /aosp_15_r20/external/libtextclassifier/native/annotator/ |
| H A D | annotator_jni.cc | 634 const CodepointSpan input_indices = in TC3_JNI_METHOD() local
665 const CodepointSpan input_indices = in TC3_JNI_METHOD() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | reorder_reshape_transpose.cc | 62 std::vector<int> input_indices; in ComputeNewPerm() local
|
| H A D | resolve_constant_unary.cc | 68 std::vector<int> input_indices = ReverseOffset(input_shape, input_offset); in ReduceGeneric() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| H A D | quantization_driver.cc | 995 Operation *op, int bias_index, const std::vector<int> &input_indices, in ShouldCheckBiasScale()
1030 Operation *op, int bias_index, const std::vector<int> &input_indices, in SetBiasParamsWithAdjustments()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseUnaryOps.cpp | 125 auto input_indices = input_impl->indices(); in coalesced_unary_ufunc_out() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_gemm_template.py | 515 input_indices=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| H A D | post_quantize.cc | 261 std::vector<uint64_t> input_indices(num_dimensions); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | sparse_ops.cc | 613 ShapeHandle input_indices = c->input(0); in __anone91b66371802() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| H A D | conv_emitter.cc | 297 std::vector<mlir::AffineExpr> input_indices; in CreateNaiveMlirConv() local
|