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

  /external/tensorflow/tensorflow/compiler/xla/
shape_util.cc 646 /* static */ int64 ShapeUtil::GetDimension(const Shape& shape,
    [all...]
  /external/tensorflow/tensorflow/core/util/
mkl_util.h 156 size_t GetDimension() const { return dimension_; }
426 inline size_t GetDimension(char dimension) const {
428 CHECK(index >= 0 && index < this->GetDimension())
449 inline size_t GetDimension() const { return data_.dimension_; }
684 for (size_t j = 0; j < mkl_shape.GetDimension(); j++) {
808 second_shape.AddDim(SIZE_OF_MKL_SERIAL_DATA(mkl_shape.GetDimension()));
842 second_shape.AddDim(SIZE_OF_MKL_SERIAL_DATA(mkl_shape.GetDimension()));
    [all...]

Completed in 528 milliseconds