Home
last modified time | relevance | path

Searched defs:inferred_shape (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Doperand_upcaster.cc50 const Shape& inferred_shape = status_or_inferred_shape.ValueOrDie().value(); in InstructionMatchesPattern() local
Dshape_inference_test.cc442 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
488 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
534 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
814 Shape inferred_shape = inferred_status.ValueOrDie(); in Pass() local
2085 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
2103 Shape inferred_shape = inferred_status.ValueOrDie(); in TEST_F() local
2174 Shape inferred_shape = inferred_shape_and_status.ValueOrDie(); in TEST_F() local
2184 Shape inferred_shape = inferred_shape_and_status.ValueOrDie(); in TEST_F() local
2396 Shape inferred_shape = statusor.ValueOrDie(); in TEST_F() local
Dshape_inference.cc2930 Shape inferred_shape = ShapeUtil::MakeShape( in InferDynamicReshapeShape() local
2948 Shape inferred_shape = in InferReshapeShape() local
Dhlo_verifier.cc1175 const Shape& inferred_shape, in CheckShape()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.cc336 [](const InferredShape& inferred_shape) { in PerformStaticShapeInferenceBeforeEncapsulation()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc653 for (const auto& inferred_shape : output_shapes) { in GetGraph() local
676 for (const auto& inferred_shape : output_shapes) { in GetGraph() local
/external/tensorflow/tensorflow/python/lib/core/
Dpy_seq_tensor.cc76 gtl::InlinedVector<int64, 4> inferred_shape; member
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc81 for (const auto& inferred_shape : output_shapes) { in ConvertGraphShapeInfoToShapeMap() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1200 bool CompatibleShapes(ShapeHandle inferred_shape, in CompatibleShapes()
1229 bool SameShapes(ShapeHandle inferred_shape, in SameShapes()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc1021 const PartialTensorShape& inferred_shape, int replica_id, in CreateConcatNodesForRetval()