Searched refs:indices0 (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | dynamic_stitch_op.cc | 61 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/ |
D | dynamic_stitch_op.cc | 196 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/ |
D | sparse_ops_test.py | 234 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/ |
D | builder_misc.cpp | 654 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()
|