Home
last modified time | relevance | path

Searched defs:TensorShape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtensor_shape.h38 class TensorShape; variable
Dallocator.h35 class TensorShape; variable
Dtensor_shape.cc831 MAKE_SHAPE(int32, TensorShape) in MAKE_SHAPE() argument
/external/llvm-project/llvm/lib/Analysis/
DTFUtils.cpp201 std::vector<int64_t> TensorShape; in getTensorSpecFromJSON() local
/external/tensorflow/tensorflow/python/framework/
Dtensor_shape.py730 class TensorShape(object): class
/external/tensorflow/tensorflow/python/tpu/
Dtpu.py1000 padded_shapes: List[Optional[tensor_shape.TensorShape]],