HomeSort by relevance Sort by last modified time
    Searched defs:buffer_shape (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
scatter_nd_op.cc 31 // buffer_shape[num_index_dims:]
32 Status ValidateUpdateShape(const TensorShape& buffer_shape,
47 "buffer_shape[num_index_dims:], got updates.shape: ",
50 ", buffer_shape: ", buffer_shape.DebugString(),
55 if (buffer_shape.dims() <
60 batch_dim + buffer_shape.dims() - num_index_dims) {
70 buffer_shape.dim_size(d + num_index_dims)) {
87 TensorShape buffer_shape; variable
88 OP_REQUIRES_OK(context, context->ConstantInputAsShape(2, &buffer_shape));
    [all...]
segment_reduction_ops.cc 77 TensorShape buffer_shape = data_shape; variable
78 buffer_shape.RemoveDimRange(0, indices_shape.dims());
79 buffer_shape.InsertDim(0, num_segments);
81 xla::Broadcast(InitialValue(builder), buffer_shape.dim_sizes());
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_compiler.cc 678 xla::Shape buffer_shape; local
680 TensorShapeToXLAShape(arg.type, shape, &buffer_shape));
682 {buffer_shape, xla::ShapeUtil::MakeShape(xla::S32, {})});
    [all...]

Completed in 324 milliseconds