Searched defs:idx_type (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | listdiff_op.cc | 50 DataType idx_type = context->expected_output_dtype(1); in Compile() local
102 Status ListDiffWithIndexType(XlaOpKernelContext* context, DataType idx_type) { in ListDiffWithIndexType()
|
| /external/mesa3d/src/gallium/drivers/freedreno/a5xx/ |
| D | fd5_draw.h | 28 enum a4xx_index_size idx_type, uint32_t max_indices, in fd5_draw()
70 enum a4xx_index_size idx_type; in fd5_draw_emit() local
|
| /external/mesa3d/src/gallium/drivers/freedreno/ |
| D | freedreno_draw.h | 29 enum pc_di_index_size idx_type, uint32_t idx_size, uint32_t idx_offset, in fd_draw()
132 enum pc_di_index_size idx_type = INDEX_SIZE_IGN; in fd_draw_emit() local
|
| /external/mesa3d/src/gallium/drivers/freedreno/a4xx/ |
| D | fd4_draw.h | 34 enum a4xx_index_size idx_type, uint32_t max_indices, in fd4_draw()
76 enum a4xx_index_size idx_type; in fd4_draw_emit() local
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | lu_op.cc | 176 #define REGISTER_LU(type, idx_type) \ argument
|
| D | lu_op_gpu.cu.cc | 260 #define REGISTER_LU_GPU(type, idx_type) \ argument
|
| /external/executorch/kernels/portable/cpu/util/ |
| D | advanced_index_util.cpp | 126 ScalarType idx_type = index.scalar_type(); in query_integral_index() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 1172 ArrayRef<int64_t> shape, Type idx_type, in BuildReshapeOp()
1183 Value begin, ArrayRef<int64_t> shape, Type idx_type, in BuildSliceOp()
1211 Type idx_type = start_indices_type.getElementType(); in matchAndRewrite() local
|