Home
last modified time | relevance | path

Searched defs:slice_shape (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/
Dsharding_util_ops.cc180 const Eigen::DSizes<Eigen::DenseIndex, Rank>& slice_shape, in GetSliceIndices()
188 const Eigen::DSizes<Eigen::DenseIndex, 1>& slice_shape, const int index) { in GetSliceIndices()
197 const Eigen::DSizes<Eigen::DenseIndex, 2>& slice_shape, const int index) { in GetSliceIndices()
207 const Eigen::DSizes<Eigen::DenseIndex, 3>& slice_shape, const int index) { in GetSliceIndices()
220 const Eigen::DSizes<Eigen::DenseIndex, 4>& slice_shape, const int index) { in GetSliceIndices()
237 const Eigen::DSizes<Eigen::DenseIndex, 5>& slice_shape, const int index) { in GetSliceIndices()
258 const Eigen::DSizes<Eigen::DenseIndex, 6>& slice_shape, const int index) { in GetSliceIndices()
284 const Eigen::DSizes<Eigen::DenseIndex, 7>& slice_shape, const int index) { in GetSliceIndices()
315 const Eigen::DSizes<Eigen::DenseIndex, 8>& slice_shape, const int index) { in GetSliceIndices()
692 const TensorShape& slice_shape = inputs[0].shape(); in GetInputsAndOutputShape() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsharding_util_ops.cc98 absl::Span<const int64> slice_shape, in GetSliceIndices()
184 std::vector<int64_t> slice_shape(rank); in CompileInternal() local
333 auto slice_shape = input_shapes[0].dim_sizes(); in CompileInternal() local
403 const TensorShape& slice_shape = input_shapes[0]; in GetInputsAndOutputShape() local
Dstack_ops.cc152 TensorShape slice_shape = elem_shape; in Compile() local
208 auto slice_shape = stack_shape.dim_sizes(); in Compile() local
Dtensor_list_utils.cc442 std::vector<int64_t> slice_shape = in ExecuteTensorListPopBack() local
523 std::vector<int64_t> slice_shape = in ExecuteTensorListGetItem() local
Dtensor_array_ops.cc219 TensorShape slice_shape = elem_shape; in Compile() local
270 auto slice_shape = ta_shape.dim_sizes(); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_fusion_test.cc69 Shape slice_shape = ShapeUtil::MakeShape(F32, {10, 2}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dall_to_all_decomposer.cc100 Shape slice_shape = all_to_all->shape(); in ExpandInstruction() local
Ddynamic_padder.cc1471 Shape slice_shape = static_shape; in RewriteDynamicBinaryOp() local
Dalgebraic_simplifier_test.cc4719 Shape slice_shape = ShapeUtil::MakeShape(F32, {2, 2, 3, 3}); in TEST_F() local
5047 Shape slice_shape = ShapeUtil::MakeShape(F32, {10, 1, 1000}); in TEST_F() local
8383 const Shape& slice_shape = in TEST_F() local
Dhlo_dataflow_analysis_test.cc2298 Shape slice_shape = ShapeUtil::MakeShape(F32, {1, 2}); in TEST_F() local
/external/ComputeLibrary/tests/validation/fixtures/
DUnstackFixture.h102 …const TensorShape slice_shape = arm_compute::misc::shape_calculator::calculate_unstack_… in compute_reference() local
/external/tensorflow/tensorflow/core/kernels/
Dload_and_remap_matrix_op.cc214 TensorShape slice_shape; in Compute() local
Dsave_restore_v2_ops.cc124 TensorShape slice_shape; in Compute() local
Dsave_restore_tensor.cc113 TensorShape slice_shape; in SaveTensors() local
Dscatter_nd_op.cc1020 auto slice_shape = indices.shape(); in DoScatterNd() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc371 Shape slice_shape = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
448 Shape slice_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
536 Shape slice_shape = ShapeUtil::MakeShape(F32, {10, 1, 1000}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_padding_legalization.cc382 Shape slice_shape = in CanonicalizeBackwardInputConvolution() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc93 auto slice_shape = concat->shape(); in PadEachPartitionWithHaloExchange() local
Dspmd_partitioner_util.cc403 auto slice_shape = concat->shape(); in TileToPartialReplicateHaloExchange() local
514 auto slice_shape = concat->shape(); in PadFromPartialReplicateShape() local
952 auto slice_shape = concat->shape(); in ExchangeHaloAndGetValidData() local
Ddot_handler.cc1357 auto slice_shape = dot->shape(); in PartitionBaseCase() local
Dspmd_partitioner.cc3788 const Shape& slice_shape = ShapeUtil::GetSubshape( in HandleOutfeed() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcollection_ops_util.cc106 auto slice_shape = llvm::to_vector<8>(buffer_type.getShape()); in SetElement() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc623 const Shape& slice_shape = use->opcode() == HloOpcode::kDynamicSlice in CreateLiteralForConstrainedUses() local
/external/python/cpython3/Lib/test/
Dtest_buffer.py357 def slice_shape(lst, slices): function
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3340 auto slice_shape = llvm::to_vector<4>(input_type.getShape()); in matchAndRewrite() local
5573 SmallVector<int64_t, 4> slice_shape(value_rank, ShapedType::kDynamicSize); in matchAndRewrite() local