Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_ops.cc227 auto slice_shape = shape.dim_sizes(); in Compile() local
366 TensorShape slice_shape = elem_shape; in Compile() local
409 TensorShape slice_shape = elem_shape; in Compile() local
453 auto slice_shape = shape.dim_sizes(); in Compile() local
Dstack_ops.cc152 TensorShape slice_shape = elem_shape; in Compile() local
208 auto slice_shape = stack_shape.dim_sizes(); in Compile() 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/core/kernels/
Dload_and_remap_matrix_op.cc205 TensorShape slice_shape; in Compute() local
Dsave_restore_v2_ops.cc120 TensorShape slice_shape; in Compute() local
Dsave_restore_tensor.cc107 TensorShape slice_shape; in SaveTensors() local
Dscatter_nd_op.cc726 auto slice_shape = indices.shape(); in DoScatterNd() local
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
Dshape.py355 def slice_shape(start_sum, size, name): function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc387 Shape slice_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
464 Shape slice_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F() local
552 Shape slice_shape = ShapeUtil::MakeShape(F32, {10, 1, 1000}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_padding_legalization.cc370 Shape slice_shape = in CanonicalizeBackwardInputConvolution() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dtest_utils.cc450 const Shape& slice_shape = use->opcode() == HloOpcode::kDynamicSlice in CreateLiteralForConstrainedUses() local
/external/python/cpython3/Lib/test/
Dtest_buffer.py349 def slice_shape(lst, slices): function
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dataflow_analysis_test.cc2180 Shape slice_shape = ShapeUtil::MakeShape(F32, {1, 2}); in TEST_F() local
Dalgebraic_simplifier_test.cc3515 Shape slice_shape = ShapeUtil::MakeShape(F32, {2, 2, 3, 3}); in TEST_F() local
3841 Shape slice_shape = ShapeUtil::MakeShape(F32, {10, 1, 1000}); in TEST_F() local