Searched defs:symbolic_shapes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/runtime/ |
| D | symbolic_shape_test.cc | 649 ArrayRef<SymbolicShape> symbolic_shapes) { in HashSymbolicShapes()
657 llvm::SmallVector<SymbolicShape> symbolic_shapes(1, {1}); in BM_Hash1x1() local
662 llvm::SmallVector<SymbolicShape> symbolic_shapes(4, {1}); in BM_Hash1x4() local
667 llvm::SmallVector<SymbolicShape> symbolic_shapes(8, {1}); in BM_Hash1x8() local
672 llvm::SmallVector<SymbolicShape> symbolic_shapes(1, {1, 2}); in BM_Hash2x1() local
677 llvm::SmallVector<SymbolicShape> symbolic_shapes(4, {1, 2}); in BM_Hash2x4() local
682 llvm::SmallVector<SymbolicShape> symbolic_shapes(8, {1, 2}); in BM_Hash2x8() local
|
| D | symbolic_shape.cc | 140 ArrayRef<size_t> iteration_order, SymbolicShapes& symbolic_shapes) { in ResolveImpl()
224 llvm::SmallVector<SymbolicShape> symbolic_shapes; in Resolve() local
296 ArrayRef<SymbolicShape> symbolic_shapes) { in Hash()
|
| D | jit_executable.cc | 294 symbolic_shapes = symbolic_shapes_resolver_.Resolve(arguments); in GetExecutable() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/ |
| D | specialization.cc | 144 ArrayRef<SymbolicShape> symbolic_shapes, in SpecializeFunction()
|
| D | jit_compiler.cc | 279 ArrayRef<SymbolicShape> symbolic_shapes, in Specialize()
|