OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:indices0
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
dynamic_stitch_op.cc
60
// Check if data0.shape[
indices0
.dims():] == data1.shape[indices1.dims():]
61
static bool SameExtraShape(const Tensor& data0, const Tensor&
indices0
,
63
const int extra0 = data0.dims() -
indices0
.dims();
67
if (data0.dim_size(
indices0
.dims() + i) !=
103
const Tensor&
indices0
= (*indices_inputs)[0];
local
114
c, input_num == 0 || SameExtraShape(data0,
indices0
, data, indices),
116
"Need data[0].shape[",
indices0
.dims(), ":] = data[", input_num,
120
", indices[0].shape = ",
indices0
.shape().DebugString(),
129
for (int d =
indices0
.dims(); d < data0.dims(); d++) {
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
dynamic_stitch_op.cc
185
// Check if data0_shape[
indices0
.dims():] == data1_shape[indices1.dims():]
187
const TensorShape&
indices0
,
190
const int extra0 = data0_shape.dims() -
indices0
.dims();
194
if (data0_shape.dim_size(
indices0
.dims() + i) !=
/external/tensorflow/tensorflow/python/kernel_tests/
sparse_ops_test.py
216
indices0
= np.array([0, 13, 10, 33, 32, 14])
220
indices0
= sparse_tensor.SparseTensorValue(
222
np.array(
indices0
, indices_dtype), np.array(shape, np.int64))
229
return ([sparse_tensor.SparseTensor.from_value(
indices0
),
[
all
...]
Completed in 875 milliseconds