| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.cc | 45 Status ExecutionInput::SetDynamicShape(Shape dynamic_shape) { in SetDynamicShape() argument
47 if (!ShapeUtil::DynamicShapeIsCompatible(input_shape, dynamic_shape)) { in SetDynamicShape()
50 dynamic_shape.DebugString()); in SetDynamicShape()
52 dynamic_shape_ = absl::make_unique<Shape>(std::move(dynamic_shape)); in SetDynamicShape()
|
| D | executable.h | 97 Status SetDynamicShape(Shape dynamic_shape);
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_util.cc | 93 for (auto& dynamic_shape : dynamic_shapes) { in DynamicShapesToTensorShapes() local
95 ShapeTensorToTensorShape(dynamic_shape.tensor(), &(*shapes)[i])); in DynamicShapesToTensorShapes()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 167 bool DynamicShapeIsCompatible(const xla::Shape& dynamic_shape, in DynamicShapeIsCompatible() argument
169 if (dynamic_shape.rank() != bounded_shape.rank()) { in DynamicShapeIsCompatible()
172 for (int64 i = 0; i < dynamic_shape.rank(); ++i) { in DynamicShapeIsCompatible()
173 if (dynamic_shape.dimensions(i) > bounded_shape.dimensions(i)) { in DynamicShapeIsCompatible()
|
| D | tpu_on_demand_compiler.cc | 122 ApiConverter::ToC(arg.shape(), &se_args[i]->dynamic_shape); in ExecuteAsyncOnStream()
144 ApiConverter::Free(&se_args[i]->dynamic_shape); in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tensordot_op_test.py | 245 for dynamic_shape in set([False, True]):
247 num_dims, dynamic_shape):
250 dynamic_shape)
|
| D | sparse_ops_test.py | 213 dynamic_shape = array_ops.placeholder_with_default(shape, shape=[2])
217 dense_shape=dynamic_shape)
221 dense_shape=dynamic_shape)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | padding_fifo_queue.cc | 162 std::vector<bool> dynamic_shape; in TryDequeueMany()
197 dynamic_shape.push_back(has_dynamic_shape); in TryDequeueMany()
205 if (dynamic_shape[i]) { in TryDequeueMany()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1576 const xla::Shape& dynamic_shape, const xla::Shape& bounded_shape) { in DynamicArrayShapeIsCompatible() argument
1577 if (dynamic_shape.rank() != bounded_shape.rank()) { in DynamicArrayShapeIsCompatible()
1580 for (int64 i = 0; i < dynamic_shape.rank(); ++i) { in DynamicArrayShapeIsCompatible()
1581 if (dynamic_shape.dimensions(i) > bounded_shape.dimensions(i)) { in DynamicArrayShapeIsCompatible()
1589 const xla::Shape& dynamic_shape, const xla::Shape& bounded_shape) { in DynamicShapeIsCompatible() argument
1591 xla::ShapeUtil::ForEachSubshape(dynamic_shape, [&](const Shape& sub_shape, in DynamicShapeIsCompatible()
|
| D | shape_util.h | 697 static bool DynamicArrayShapeIsCompatible(const xla::Shape& dynamic_shape,
701 static bool DynamicShapeIsCompatible(const xla::Shape& dynamic_shape,
|
| D | literal_test.cc | 758 auto dynamic_shape = ShapeUtil::MakeShape(F32, {2, 3}, {false, true}); in TEST_F() local
759 auto dynamic_literal = original.ToBoundedDynamic(dynamic_shape); in TEST_F()
760 EXPECT_EQ(dynamic_literal.shape(), dynamic_shape); in TEST_F()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_decl.h | 259 XLA_Shape dynamic_shape; member
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | hlo_legalize_to_lhlo.cc | 351 SmallVector<int64_t, 2> dynamic_shape(result_rank, in InsertDynamicMemrefCastOp() local
354 dynamic_shape, operand_type.getElementType(), in InsertDynamicMemrefCastOp()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | check_ops.py | 1584 dynamic_shape = array_ops.shape(x)
1592 int(size) if size is not None else dynamic_shape[i]
1598 has_rank_zero, lambda: array_ops.constant([1]), lambda: dynamic_shape)
|
| D | image_ops_impl.py | 118 dynamic_shape = array_ops.unstack(array_ops.shape(image), rank)
120 s if s is not None else d for s, d in zip(static_shape, dynamic_shape)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 565 xla::XlaOp dynamic_shape = ctx->Input(0); in Compile() local
577 auto dim_size = xla::Slice(dynamic_shape, {dim}, {dim + 1}, {1}); in Compile()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy_test.py | 413 dynamic_shape = array_ops.shape(input=tensor)
415 shape[index] = dynamic_shape[index]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1367 llvm::SmallVector<int64_t, 8> dynamic_shape(shape_ty.getDimSize(0), in GetReshapeOutputType() local
1369 output_ty = RankedTensorType::get(dynamic_shape, element_ty); in GetReshapeOutputType()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 595 llvm::SmallVector<int64_t, 8> dynamic_shape(shape_ty.getDimSize(0), in GetReshapeOutputType() local
597 output_ty = RankedTensorType::get(dynamic_shape, element_ty); in GetReshapeOutputType()
|