Searched refs:indices1 (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | dynamic_stitch_op.cc | 198 const TensorShape& indices1) { in SameExtraShape() argument
200 const int extra1 = data1_shape.dims() - indices1.dims(); in SameExtraShape()
204 data1_shape.dim_size(indices1.dims() + i)) { in SameExtraShape()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | toco_convert_test.cc | 146 name: "indices1" in TEST()
154 input: "indices1" in TEST()
|
| /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 | 211 indices1 = array_ops.placeholder(dtypes.int32, shape=None)
216 ragged_gather_ops.gather_nd(params, indices1)
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_morphing.cpp | 145 Eigen::DSizes<ptrdiff_t, 2> indices1(1,2); in test_slice_in_expr() local
149 …tensor3 = tensor1.slice(indices1, sizes1).contract(tensor2.slice(indices2, sizes2), contract_along… in test_slice_in_expr()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_ops_test.py | 235 indices1 = np.array([12, 4, 0, 0, 1, 30])
241 indices1 = sparse_tensor.SparseTensorValue(
243 np.array(indices1, indices_dtype), np.array(shape, np.int64))
248 sparse_tensor.SparseTensor.from_value(indices1)],
|
| /external/mesa3d/src/gallium/drivers/swr/rasterizer/jitter/ |
| D | builder_misc.cpp | 655 Value *indices1 = EXTRACT_16(vIndices, 1); in GATHERPS_16() local
661 Value *gather1 = GATHERPS(src1, pBase, indices1, mask1, scale); in GATHERPS_16()
737 Value *indices1 = EXTRACT_16(vIndices, 1); in GATHERDD_16() local
743 Value *gather1 = GATHERDD(src1, pBase, indices1, mask1, scale); in GATHERDD_16()
|