| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.cc | 116 mlir::ModuleOp module, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in GetXlaInputShapes()
326 Status RefineShapes(llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in RefineShapes()
554 llvm::ArrayRef<TensorOrResourceShape> arg_shapes) { in CompileMlirSetup()
567 llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in BuildHloFromTf()
615 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in PopulateResultIOInfo()
636 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirToXlaHlo()
666 llvm::StringRef mlir_module_string, llvm::ArrayRef<TensorShape> arg_shapes, in CompileSerializedMlirToXlaHlo()
767 llvm::SmallVector<TensorOrResourceShape, 4>& arg_shapes) { in CompileGraphSetup()
809 llvm::SmallVector<TensorOrResourceShape, 4> arg_shapes; in BuildHloFromModule() local
828 llvm::SmallVector<TensorOrResourceShape, 4> arg_shapes; in CompileGraphToXlaHlo() local
|
| D | tf_xla_mlir_translate.cc | 133 llvm::SmallVectorImpl<TensorOrResourceShape>& arg_shapes) { in ParseArgumentShapes()
255 mlir::ModuleOp module_op, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in CompileMlirToXlaHloViaBuilder()
313 llvm::SmallVector<TensorOrResourceShape, 4> arg_shapes; in MlirTfToHloTextTranslateFunctionImpl() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_compile.cc | 165 const std::vector<PartialTensorShape>& arg_shapes, in OptimizeGraph()
253 const std::vector<TensorShape>& arg_shapes, in BuildComputationArgumentDescriptions()
340 const std::vector<PartialTensorShape>& arg_shapes, Graph* graph, in RunShapeInferenceOnComputation()
374 const std::vector<TensorShape>& arg_shapes, in CompileTFFunctionToHlo()
453 absl::Span<const TensorShape> arg_shapes, in GetShardingInfo()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile.proto | 45 repeated TensorShapeProto arg_shapes = 6; field
|
| D | tpu_compile_op_support.cc | 168 std::vector<Shape> arg_shapes; in GetPerDeviceShape() local
345 const std::vector<TensorShape>& arg_shapes) { in CreateTpuCompilationRequest()
449 std::vector<TensorShape>* arg_shapes) { in ComputeArgumentShapes()
|
| D | tpu_compile_op_impl.cc | 31 const std::vector<TensorShape>& arg_shapes, in Compile()
|
| D | tpu_compile_op_common.cc | 185 std::vector<TensorShape> arg_shapes; in CompileLocallyAndFillHostCacheInternal() local
468 const std::vector<TensorShape>& arg_shapes, in RegisterXLAFingerprints()
|
| D | tpu_functional_ops.cc | 402 std::map<int, InferredShape>& arg_shapes, EdgeShapes& tpu_input_shapes, in GetInputOutputInfo()
531 std::vector<TensorShape> arg_shapes; in MaybeRegisterFingerprint() local
1997 std::map<int, InferredShape>& arg_shapes, in InferShapesWithResourceVar()
2197 std::map<int, InferredShape> arg_shapes; in OptimizeTpuInputOutputTensors() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference.cc | 46 const std::map<int, InferredShape>& arg_shapes, in PropagateShapes()
243 Status InferShapes(Graph* graph, const std::map<int, InferredShape>& arg_shapes, in InferShapes()
|
| D | encapsulate_util.cc | 323 std::map<int, InferredShape> arg_shapes; in PerformStaticShapeInferenceBeforeEncapsulation() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 48 const std::vector<TensorShape> arg_shapes; member
279 std::vector<TensorShape> arg_shapes; in SetupArgs() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function.cc | 111 std::vector<const xla::Shape*> arg_shapes; in Compile() local
|
| D | xla_compiler.cc | 1050 std::vector<xla::Shape> arg_shapes(input_to_args->size()); in BuildArguments() local
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | utils.cc | 78 std::vector<Shape> arg_shapes; in GetShardedProgramShapes() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 364 absl::Span<const Shape* const> arg_shapes, const int64_t dimension) { in InferConcatOpShape()
1115 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply, in InferMapShape()
2217 absl::Span<const Shape* const> arg_shapes, in InferReduceShape()
3298 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply) { in InferCallShape()
3658 absl::Span<const Shape* const> arg_shapes, in InferScatterShape()
|
| D | hlo_parser.cc | 1782 absl::InlinedVector<const Shape*, 2> arg_shapes; in CreateInstruction() local
1801 absl::InlinedVector<const Shape*, 2> arg_shapes; in CreateInstruction() local
1935 absl::InlinedVector<const Shape*, 2> arg_shapes; in CreateInstruction() local
2143 absl::InlinedVector<const Shape*, 2> arg_shapes; in CreateInstruction() local
2168 absl::InlinedVector<const Shape*, 2> arg_shapes; in CreateInstruction() local
2199 absl::InlinedVector<const Shape*, 2> arg_shapes; in CreateInstruction() local
2890 absl::InlinedVector<const Shape*, 3> arg_shapes; in CreateInstruction() local
|
| D | hlo_verifier.cc | 1663 absl::InlinedVector<const Shape*, 3> arg_shapes; in HandleScatter() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1907 const ParameterInfo& params_info, std::vector<InferredShape>* arg_shapes, in GetArgAndRetvalShapes()
2127 const std::vector<InferredShape>& arg_shapes, in AssignArgsAndRetvalsToCores()
2464 const Node& replicate_node, const std::vector<InferredShape>& arg_shapes, in BuildDynamicShapeNodes()
2595 const std::vector<InferredShape>& arg_shapes, in BuildCompileNode()
3158 const std::vector<InferredShape>& arg_shapes, in CreateOrGetPerHostVariableCopy()
3243 const std::vector<InferredShape>& arg_shapes, in BuildExecuteNodes()
4754 std::vector<InferredShape> arg_shapes, retval_shapes; in RewriteTPUReplicateNode() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 2322 llvm::SmallVectorImpl<xla::Shape>* arg_shapes, in SetEntryTupleShapesAndLeafReplication()
2359 llvm::SmallVectorImpl<xla::Shape>* arg_shapes) { in SetEntryTupleShardings()
2399 llvm::SmallVector<xla::Shape, 4> arg_shapes; in LowerBasicBlockAsFunction() local
2426 llvm::SmallVector<xla::Shape, 4> arg_shapes; in LowerBasicBlockAsFunction() local
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 946 arg_shapes=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 2684 ArrayRef<ArrayRef<int64_t>> arg_shapes, in InferShapeForFunction()
|