Searched defs:arg_shapes (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference.cc | 43 const std::map<int, InferredShape>& arg_shapes, in PropagateShapes()
135 Status InferShapes(Graph* graph, const std::map<int, InferredShape>& arg_shapes, in InferShapes()
|
| D | xla_compilation_cache.h | 100 std::vector<std::pair<DataType, std::vector<int64>>> arg_shapes; member
|
| D | encapsulate_util.cc | 315 std::map<int, InferredShape> arg_shapes; in PerformStaticShapeInferenceBeforeEncapsulation() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | index_ops_cpu.cc | 108 std::vector<xla::Shape> arg_shapes; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_jit_compiled_cpu_function.cc | 104 std::vector<const xla::Shape*> arg_shapes; in Compile() local
|
| D | xla_compiler.cc | 767 std::vector<xla::Shape> arg_shapes(input_to_args->size()); in BuildArguments() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 138 const std::vector<TensorShape> arg_shapes; member
287 std::vector<TensorShape> arg_shapes; in SetupArgs() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 349 absl::Span<const Shape* const> arg_shapes, const int64 dimension) { in InferConcatOpShape()
1073 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply, in InferMapShape()
2043 absl::Span<const Shape* const> arg_shapes, in InferReduceShape()
2834 absl::Span<const Shape* const> arg_shapes, const ProgramShape& to_apply) { in InferCallShape()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | numpy_bridge.cc | 157 Safe_PyObjectPtr arg_shapes = make_safe(PyTuple_New(shape.parameters_size())); in PyProgramShapeInfoFromXlaProgramShape() local
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | function.py | 844 arg_shapes=None, argument
|