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

  /external/tensorflow/tensorflow/core/kernels/
reshape_util.cc 61 auto target_shape = target_shape_in.vec<int64>(); local
63 const int64 size = target_shape(d);
  /external/tensorflow/tensorflow/core/util/
tensor_slice_set.cc 80 TensorShape target_shape; local
82 s = slice.SliceTensorShape(shape_, &target_shape);
87 int64 total_size = target_shape.num_elements();
137 TensorShape target_shape; local
139 s = slice.SliceTensorShape(shape_, &target_shape);
144 int64 total_size = target_shape.num_elements();
  /external/tensorflow/tensorflow/core/grappler/optimizers/
arithmetic_optimizer_test.cc 617 Output target_shape = ops::Concat( local
618 s.WithOpName("target_shape"),
620 Output reshape = ops::Reshape(s, inputs, target_shape);
    [all...]

Completed in 283 milliseconds