Home
last modified time | relevance | path

Searched defs:dynamic_shapes (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc99 OpInputList dynamic_shapes; in GetDynamicShapes() local
166 const std::vector<TensorShape>& dynamic_shapes, in CompileLocallyAndFillHostCache()
181 const std::vector<TensorShape>& dynamic_shapes, in CompileLocallyAndFillHostCacheInternal()
233 std::vector<TensorShape> dynamic_shapes; in ComputeInternal() local
Dtpu_util.cc79 Status DynamicShapesToTensorShapes(const OpInputList& dynamic_shapes, in DynamicShapesToTensorShapes()
89 Status DynamicShapesToTensorShapes(const InputList& dynamic_shapes, in DynamicShapesToTensorShapes()
Dtpu_op_util.cc30 const std::vector<tensorflow::TensorShape>& dynamic_shapes) { in CreateShapePrefix()
108 const std::vector<TensorShape>& dynamic_shapes, in CreateCompilationCacheKey()
Dtpu_compile_op_common.h141 const std::vector<TensorShape>& dynamic_shapes, in LookupPersistentCompilationCacheAndFillCaches()
Dtpu_compile_op_support.cc448 const std::vector<TensorShape>& dynamic_shapes, in ComputeArgumentShapes()