Home
last modified time | relevance | path

Searched refs:indices0 (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddynamic_stitch_op.cc61 static bool SameExtraShape(const Tensor& data0, const Tensor& indices0, in SameExtraShape() argument
63 const int extra0 = data0.dims() - indices0.dims(); in SameExtraShape()
67 if (data0.dim_size(indices0.dims() + i) != in SameExtraShape()
103 const Tensor& indices0 = (*indices_inputs)[0]; in CheckArgsAndAllocateResult() local
114 c, input_num == 0 || SameExtraShape(data0, indices0, data, indices), in CheckArgsAndAllocateResult()
116 "Need data[0].shape[", indices0.dims(), ":] = data[", input_num, in CheckArgsAndAllocateResult()
120 ", indices[0].shape = ", indices0.shape().DebugString(), in CheckArgsAndAllocateResult()
129 for (int d = indices0.dims(); d < data0.dims(); d++) { in CheckArgsAndAllocateResult()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddynamic_stitch_op.cc196 const TensorShape& indices0, in SameExtraShape() argument
199 const int extra0 = data0_shape.dims() - indices0.dims(); in SameExtraShape()
203 if (data0_shape.dim_size(indices0.dims() + i) != in SameExtraShape()
/external/deqp/external/vulkancts/data/vulkan/amber/spirv_assembly/instruction/compute/variable_pointer/dynamic_offset/
Dselect_descriptor_array.amber124 BUFFER indices0 DATA_TYPE int32 DATA
143 BIND BUFFER indices0 AS push_constant
/external/mesa3d/src/gallium/drivers/swr/rasterizer/jitter/functionpasses/
Dlower_x86.cpp630 Value* indices0 = B->VSHUFFLE(vi32Indices, vi32Indices, B->C({0, 1, 2, 3})); in VGATHER_EMU() local
650 B->CALL(pX86IntrinFunc, {src0, pBase, indices0, mask0, i8Scale}); in VGATHER_EMU()
666 Value* indices0 = B->EXTRACT_16(vi32Indices, 0); in VGATHER_EMU() local
673 B->CALL(pX86IntrinFunc, {src0, pBase, indices0, mask0, i8Scale}); in VGATHER_EMU()
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparse_ops_test.py234 indices0 = np.array([0, 13, 10, 33, 32, 14])
238 indices0 = sparse_tensor.SparseTensorValue(
240 np.array(indices0, indices_dtype), np.array(shape, np.int64))
247 return ([sparse_tensor.SparseTensor.from_value(indices0),
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlower_tf.mlir630 %indices0 = "tf.Const"() {value = dense<4> : tensor<i32>} : () -> tensor<i32>
632 …%0 = "tf.DynamicStitch"(%indices0, %indices1, %arg0, %arg1) : (tensor<i32>, tensor<2x2xi32>, tenso…
Dtf-ops.mlir3394 %indices0 = "tf.Const"() {value = dense<4> : tensor<i32>} : () -> tensor<i32>
3398 …%0 = "tf.DynamicStitch"(%indices0, %indices1, %arg0, %arg1) : (tensor<i32>, tensor<2x2xi32>, tenso…
3431 %indices0 = "tf.Const"() {value = dense<1> : tensor<i32>} : () -> tensor<i32>
3435 …%0 = "tf.DynamicStitch"(%indices0, %indices1, %arg0, %arg1) : (tensor<i32>, tensor<i32>, tensor<?x…