| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_impl_3.cc | 22 name, type, index_type, initial_value_functor, reduction_functor) \ argument
34 #define REGISTER_REAL_CPU_UNSORTED_KERNELS(type, index_type) \ argument
48 #define REGISTER_COMPLEX_CPU_UNSORTED_KERNELS(type, index_type) \ argument
74 name, type, index_type, initial_value_functor, reduction_kernel_functor) \ argument
88 #define REGISTER_REAL_GPU_UNSORTED_KERNELS(type, index_type) \ argument
99 #define REGISTER_SUM_GPU_UNSORTED_KERNELS(type, index_type) \ argument
|
| D | segment_reduction_ops_impl_4.cc | 22 name, type, index_type, initial_value_functor, reduction_functor) \ argument
34 #define REGISTER_REAL_CPU_UNSORTED_KERNELS(type, index_type) \ argument
48 #define REGISTER_COMPLEX_CPU_UNSORTED_KERNELS(type, index_type) \ argument
74 name, type, index_type, initial_value_functor, reduction_kernel_functor) \ argument
88 #define REGISTER_REAL_GPU_UNSORTED_KERNELS(type, index_type) \ argument
99 #define REGISTER_SUM_GPU_UNSORTED_KERNELS(type, index_type) \ argument
|
| D | segment_reduction_ops_impl_2.cc | 21 #define REGISTER_CPU_KERNEL_SEGMENT(name, functor, type, index_type, \ argument
30 #define REGISTER_REAL_CPU_KERNELS(type, index_type) \ argument
42 #define REGISTER_COMPLEX_CPU_KERNELS(type, index_type) \ argument
66 #define REGISTER_GPU_SORTED_KERNELS(type, index_type) \ argument
|
| D | segment_reduction_ops_impl_1.cc | 69 #define REGISTER_CPU_KERNEL_SEGMENT(name, functor, type, index_type, \ argument
78 #define REGISTER_REAL_CPU_KERNELS(type, index_type) \ argument
90 #define REGISTER_COMPLEX_CPU_KERNELS(type, index_type) \ argument
114 #define REGISTER_GPU_SORTED_KERNELS(type, index_type) \ argument
|
| D | one_hot_op.cc | 129 #define REGISTER_ONE_HOT_INDEX(type, index_type) \ argument
174 #define REGISTER_ONE_HOT_GPU_INDEX(type, index_type) \ argument
|
| D | scatter_nd_op.cc | 310 #define REGISTER_SCATTER_ND_KERNEL_INDEX(type, index_type, dev, name) \ argument
318 #define REGISTER_SCATTER_ND_UPDATE_KERNEL_INDEX(type, index_type, dev, name, \ argument
327 #define REGISTER_RESOURCE_SCATTER_ND_UPDATE_KERNEL_INDEX(type, index_type, \ argument
390 #define REGISTER_SCATTER_ND_TENSOR_UPDATE_TYPE_INDEX_TYPE(type, index_type, \ argument
399 #define REGISTER_SCATTER_ND_TENSOR_ADD_TYPE_INDEX_TYPE(type, index_type, dev) \ argument
407 #define REGISTER_SCATTER_ND_TENSOR_SUB_TYPE_INDEX_TYPE(type, index_type, dev) \ argument
|
| D | scatter_nd_op_test.cc | 41 void MakeOp(DataType variable_ref_type, DataType index_type) { in MakeOp()
243 void MakeBenchmarkOp(const char* op, DataType index_type) { in MakeBenchmarkOp()
|
| D | sparse_to_dense_op.cc | 142 #define REGISTER_KERNELS(type, index_type) \ argument
|
| D | scatter_op_test.cc | 41 void MakeOp(DataType variable_ref_type, DataType index_type) { in MakeOp()
52 void MakeOp(DataType variable_ref_type, DataType index_type) { in MakeOp()
272 void MakeBenchmarkOp(const char* op, DataType index_type) { in MakeBenchmarkOp()
|
| D | gather_nd_op.cc | 52 #define REGISTER_GATHER_ND_FULL(dev, type, index_type) \ argument
|
| D | gather_nd_op_test.cc | 60 void MakeOp(DataType param_type, DataType index_type) { in MakeOp()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | dynamic_slice_ops.cc | 37 DataType index_type = ctx->InputType("indices"); in Compile() local
71 DataType index_type = ctx->InputType("start_indices"); in Compile() local
|
| D | gather_op.cc | 38 DataType index_type, xla::XlaBuilder* builder, in XlaGather()
216 DataType index_type = context->input_type(1); in XlaGatherWithBatchDimsOpImpl() local
|
| D | index_ops.cc | 63 DataType index_type = output_type(0); in Compile() local
|
| D | matrix_band_part_op.cc | 54 DataType index_type = context->input_type(1); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | loop_emitter.cc | 88 absl::string_view loop_name, llvm::Type* index_type) { in EmitIndexAndSetExitBasicBlock()
126 llvm::Type* index_type) { in EmitLoop()
|
| D | ir_array.cc | 34 llvm::Type* index_type) in Index()
86 llvm::Type* index_type) in Index()
91 const Shape& shape, llvm::Type* index_type) in Index()
|
| /external/skqp/src/sksl/ir/ |
| D | SkSLIndexExpression.h | 20 static const Type& index_type(const Context& context, const Type& type) { in index_type() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_loop_emitter.cc | 34 llvm::Type* index_type) { in EmitIndexAndSetExitBasicBlock()
|
| /external/skia/src/sksl/ir/ |
| D | SkSLIndexExpression.h | 20 static const Type& index_type(const Context& context, const Type& type) { in index_type() function
|
| /external/linux-kselftest/tools/testing/selftests/bpf/ |
| D | test_btf.h | 22 #define BTF_ARRAY_ENC(type, index_type, nr_elems) \ argument
24 #define BTF_TYPE_ARRAY_ENC(type, index_type, nr_elems) \ argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | parallel_loop_emitter.cc | 61 llvm::Type* index_type) { in EmitIndexAndSetExitBasicBlock()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/fuzz/ |
| D | transformation_access_chain.cpp | 205 auto index_type = in GetIndexValue() local
|
| /external/angle/third_party/spirv-tools/src/source/fuzz/ |
| D | transformation_access_chain.cpp | 207 auto index_type = in GetIndexValue() local
|
| /external/deqp-deps/SPIRV-Tools/source/fuzz/ |
| D | transformation_access_chain.cpp | 207 auto index_type = in GetIndexValue() local
|