OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:params_shape
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
gather_op.cc
221
TensorShape
params_shape
= context->InputShape(0);
variable
223
OP_REQUIRES(context, TensorShapeUtils::IsVectorOrHigher(
params_shape
),
230
context, num_index_dims <=
params_shape
.dims(),
234
params_shape
.dims()));
240
OP_REQUIRES_OK(context, XlaGather(params,
params_shape
, indices,
/external/tensorflow/tensorflow/core/kernels/
scatter_nd_op.cc
130
TensorShape
params_shape
;
local
144
params_shape
= params.shape();
147
params_shape
= params.shape();
153
params_shape
= c->input(0).shape();
154
if (!c->forward_input_to_output_with_shape(0, 0,
params_shape
,
158
OP_REQUIRES_OK(c, c->allocate_output(0,
params_shape
, ¶ms_ptr));
170
c, indices, updates,
params_shape
, ¶ms, false /*allocate*/));
293
//
params_shape
[slice_dim:]
294
Status ValidateUpdateShape(const TensorShape&
params_shape
,
303
"
params_shape
[slice_dim:], got updates.shape: "
[
all
...]
Completed in 295 milliseconds