HomeSort by relevance Sort by last modified time
    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, &params_ptr));
170 c, indices, updates, params_shape, &params, 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