Home
last modified time | relevance | path

Searched refs:indices0 (Results 1 – 4 of 4) 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/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/mesa3d/src/gallium/drivers/swr/rasterizer/jitter/
Dbuilder_misc.cpp654 Value *indices0 = EXTRACT_16(vIndices, 0); in GATHERPS_16() local
660 Value *gather0 = GATHERPS(src0, pBase, indices0, mask0, scale); in GATHERPS_16()
736 Value *indices0 = EXTRACT_16(vIndices, 0); in GATHERDD_16() local
742 Value *gather0 = GATHERDD(src0, pBase, indices0, mask0, scale); in GATHERDD_16()