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

  /external/tensorflow/tensorflow/compiler/tf2xla/
literal_util.cc 62 std::vector<xla::Shape> tensor_shapes(host_tensors.size());
69 &tensor_shapes[i]));
73 buf_ptrs, xla::ShapeUtil::MakeTupleShape(tensor_shapes));
  /external/tensorflow/tensorflow/python/tpu/
tpu_estimator.py     [all...]

Completed in 63 milliseconds