| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stack_ops.cc | 152 TensorShape slice_shape = elem_shape; in Compile() local
208 auto slice_shape = stack_shape.dim_sizes(); in Compile() local
|
| D | tensor_list_utils.cc | 440 std::vector<int64> slice_shape = in ExecuteTensorListPopBack() local
521 std::vector<int64> slice_shape = xla::SpanToVector(buffer_shape.dimensions()); in ExecuteTensorListGetItem() local
|
| D | tensor_array_ops.cc | 219 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/ |
| D | gpu_fusion_test.cc | 69 Shape slice_shape = ShapeUtil::MakeShape(F32, {10, 2}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_to_all_decomposer.cc | 99 Shape slice_shape = all_to_all->shape(); in ExpandInstruction() local
|
| D | hlo_dataflow_analysis_test.cc | 2306 Shape slice_shape = ShapeUtil::MakeShape(F32, {1, 2}); in TEST_F() local
|
| D | algebraic_simplifier_test.cc | 4283 Shape slice_shape = ShapeUtil::MakeShape(F32, {2, 2, 3, 3}); in TEST_F() local
4634 Shape slice_shape = ShapeUtil::MakeShape(F32, {10, 1, 1000}); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | load_and_remap_matrix_op.cc | 204 TensorShape slice_shape; in Compute() local
|
| D | save_restore_v2_ops.cc | 121 TensorShape slice_shape; in Compute() local
|
| D | save_restore_tensor.cc | 107 TensorShape slice_shape; in SaveTensors() local
|
| D | scatter_nd_op.cc | 971 auto slice_shape = indices.shape(); in DoScatterNd() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 389 Shape slice_shape = ShapeUtil::MakeShape(S32, {}); in TEST_F() local
466 Shape slice_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
554 Shape slice_shape = ShapeUtil::MakeShape(F32, {10, 1, 1000}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_padding_legalization.cc | 378 Shape slice_shape = in CanonicalizeBackwardInputConvolution() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 95 auto slice_shape = concat->shape(); in PadEachPartitionWithHaloExchange() local
|
| D | spmd_partitioner_util.cc | 488 auto slice_shape = concat->shape(); in TileToPartialReplicateHaloExchange() local
598 auto slice_shape = concat->shape(); in PadFromPartialReplicateShape() local
990 auto slice_shape = concat->shape(); in ExchangeHaloAndGetValidData() local
|
| D | dot_handler.cc | 1007 auto slice_shape = dot->shape(); in PartitionBaseCase() local
|
| D | spmd_partitioner.cc | 3006 const Shape& slice_shape = ShapeUtil::GetSubshape( in HandleOutfeed() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | collection_ops_util.cc | 106 auto slice_shape = llvm::to_vector<8>(buffer_type.getShape()); in SetElement() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.cc | 561 const Shape& slice_shape = use->opcode() == HloOpcode::kDynamicSlice in CreateLiteralForConstrainedUses() local
|
| /external/python/cpython3/Lib/test/ |
| D | test_buffer.py | 356 def slice_shape(lst, slices): function
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 228 llvm::SmallVector<int64_t, 8> slice_shape(slice_end.size()); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2935 auto slice_shape = llvm::to_vector<4>(input_type.getShape()); in matchAndRewrite() local
|