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_util.cc79 Status DynamicShapesToTensorShapes(const OpInputList& dynamic_shapes, in DynamicShapesToTensorShapes()
89 Status DynamicShapesToTensorShapes(const InputList& dynamic_shapes, in DynamicShapesToTensorShapes()
Dtpu_op_util.cc29 const std::vector<tensorflow::TensorShape>& dynamic_shapes) { in CreateShapePrefix()
107 const std::vector<TensorShape>& dynamic_shapes, in CreateCompilationCacheKey()
Dtpu_compile_op_common.cc82 OpInputList dynamic_shapes; in GetDynamicShapes() local
148 const std::vector<TensorShape>& dynamic_shapes, in CompileLocallyAndFillHostCache()
199 std::vector<TensorShape> dynamic_shapes; in ComputeInternal() local
Dtpu_compile_op_common.h131 const std::vector<TensorShape>& dynamic_shapes, in LookupPersistentCompilationCacheAndFillCaches()
Dtpu_compile_op_support.cc448 const std::vector<TensorShape>& dynamic_shapes, in ComputeArgumentShapes()