| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | dynamic_stitch_op.cc | 208 const TensorShape& indices1) { in SameExtraShape() argument
210 const int extra1 = data1_shape.dims() - indices1.dims(); in SameExtraShape()
214 data1_shape.dim_size(indices1.dims() + i)) { in SameExtraShape()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | toco_convert_test.cc | 153 name: "indices1" in TEST()
161 input: "indices1" in TEST()
|
| /external/deqp/external/vulkancts/data/vulkan/amber/spirv_assembly/instruction/compute/variable_pointer/dynamic_offset/ |
| D | select_descriptor_array.amber | 127 BUFFER indices1 DATA_TYPE int32 DATA
160 BIND BUFFER indices1 AS push_constant
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_stitch_op.cc | 62 const Tensor& data1, const Tensor& indices1) { in SameExtraShape() argument
64 const int extra1 = data1.dims() - indices1.dims(); in SameExtraShape()
68 data1.dim_size(indices1.dims() + i)) { in SameExtraShape()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_nd_op_test.py | 206 indices1 = array_ops.placeholder(dtypes.int32, shape=None)
211 ragged_gather_ops.gather_nd(params, indices1)
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_morphing.cpp | 167 Eigen::DSizes<ptrdiff_t, 2> indices1(1,2); in test_slice_in_expr() local
171 …tensor3 = tensor1.slice(indices1, sizes1).contract(tensor2.slice(indices2, sizes2), contract_along… in test_slice_in_expr()
490 Eigen::DSizes<ptrdiff_t, 3> indices1(1,2,3); in test_empty_slice() local
494 tensor.slice(indices1, sizes1) = slice1; in test_empty_slice()
|
| /external/angle/src/tests/gl_tests/ |
| D | DrawElementsTest.cpp | 681 const GLubyte indices1[] = {0, 0, 0, 0, 0, 0}; in TEST_P() local
683 glBufferData(GL_ELEMENT_ARRAY_BUFFER, sizeof(indices1), indices1, GL_STATIC_DRAW); in TEST_P()
|
| /external/mesa3d/src/gallium/drivers/swr/rasterizer/jitter/functionpasses/ |
| D | lower_x86.cpp | 631 Value* indices1 = B->VSHUFFLE(vi32Indices, vi32Indices, B->C({4, 5, 6, 7})); in VGATHER_EMU() local
654 B->CALL(pX86IntrinFunc, {src1, pBase, indices1, mask1, i8Scale}); in VGATHER_EMU()
667 Value* indices1 = B->EXTRACT_16(vi32Indices, 1); in VGATHER_EMU() local
675 B->CALL(pX86IntrinFunc, {src1, pBase, indices1, mask1, i8Scale}); in VGATHER_EMU()
|
| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| D | sparse_ops_test.py | 231 indices1 = np.array([12, 4, 0, 0, 1, 30])
237 indices1 = sparse_tensor.SparseTensorValue(
239 np.array(indices1, indices_dtype), np.array(shape, np.int64))
244 sparse_tensor.SparseTensor.from_value(indices1)],
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | lower_tf.mlir | 696 …%indices1 = "tf.Const"() {value = dense<[[3, 2], [1, 0]]> : tensor<2x2xi32>} : () -> tensor<2x2xi3…
697 …%0 = "tf.DynamicStitch"(%indices0, %indices1, %arg0, %arg1) : (tensor<i32>, tensor<2x2xi32>, tenso…
|
| D | tf-ops.mlir | 3499 …%indices1 = "tf.Const"() {value = dense<[[3, 2], [1, 0]]> : tensor<2x2xi32>} : () -> tensor<2x2xi3…
3502 …%0 = "tf.DynamicStitch"(%indices0, %indices1, %arg0, %arg1) : (tensor<i32>, tensor<2x2xi32>, tenso…
3536 %indices1 = "tf.Const"() {value = dense<0> : tensor<i32>} : () -> tensor<i32>
3539 …%0 = "tf.DynamicStitch"(%indices0, %indices1, %arg0, %arg1) : (tensor<i32>, tensor<i32>, tensor<?x…
|