HomeSort by relevance Sort by last modified time
    Searched refs:value_shape (Results 1 - 16 of 16) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
fill_op.cc 36 const TensorShape value_shape = ctx->InputShape(1); variable
41 OP_REQUIRES(ctx, IsLegacyScalar(value_shape),
43 value_shape.DebugString()));
60 if (value_shape.dims() > 0) {
61 CHECK_EQ(value_shape.dims(), 1);
tensor_array_ops.cc 277 std::vector<int64> value_shape(slice_shape.begin() + 1, slice_shape.end());
278 ctx->SetOutput(0, b->Reshape(read, value_shape));
365 const TensorShape value_shape = ctx->InputShape(2); variable
369 TensorShape elem_shape = value_shape;
389 if (status.ok() && num_indices == value_shape.dim_size(0)) {
402 auto slice_dims = value_shape.dim_sizes();
405 std::vector<int64> value_starts(value_shape.dims(), 0);
406 auto value_ends = value_shape.dim_sizes();
408 std::vector<int64> value_strides(value_shape.dims(), 1);
499 TensorShape value_shape = ctx->InputShape(1) variable
    [all...]