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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
reshape_op.cc 36 const TensorShape sizes_shape = ctx->InputShape(1); variable
38 OP_REQUIRES(ctx, IsLegacyVector(sizes_shape),
40 sizes_shape.DebugString()));
41 const int64 num_dims = sizes_shape.num_elements();
  /external/tensorflow/tensorflow/core/ops/
array_ops.cc     [all...]

Completed in 175 milliseconds