Searched refs:shape_spec (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 177 gtl::InlinedVector<int64, 4>* strides, StridedSliceShapeSpec* shape_spec) { in ValidateStridedSliceOp() argument
382 if (shape_spec != nullptr) { in ValidateStridedSliceOp()
383 shape_spec->output_to_sparse_mapping.clear(); in ValidateStridedSliceOp()
384 shape_spec->output_to_processing_mapping.clear(); in ValidateStridedSliceOp()
385 shape_spec->processing_to_sparse_mapping.assign( in ValidateStridedSliceOp()
389 shape_spec->begin_dense_mask = dense_spec.begin_mask; in ValidateStridedSliceOp()
390 shape_spec->end_dense_mask = dense_spec.end_mask; in ValidateStridedSliceOp()
391 shape_spec->shrink_axis_dense_mask = dense_spec.shrink_axis_mask; in ValidateStridedSliceOp()
401 if (shape_spec != nullptr) { in ValidateStridedSliceOp()
402 shape_spec->output_to_sparse_mapping.push_back(sparse_index); in ValidateStridedSliceOp()
[all …]
|
| D | strided_slice_op.h | 73 StridedSliceShapeSpec* shape_spec = nullptr);
84 StridedSliceShapeSpec* shape_spec = nullptr);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 61 const StridedSliceShapeSpec& shape_spec, in EmitDynamicSlice() argument
79 input_shape.dim_size(shape_spec.output_to_processing_mapping[i])); in EmitDynamicSlice()
114 int64 sparse_index = shape_spec.processing_to_sparse_mapping[i]; in EmitDynamicSlice()
115 bool shrink_axis_set = (1 << i) & shape_spec.shrink_axis_dense_mask; in EmitDynamicSlice()
142 bool begin_mask = (1 << i) & shape_spec.begin_dense_mask; in EmitDynamicSlice()
143 bool end_mask = (1 << i) & shape_spec.end_dense_mask; in EmitDynamicSlice()
146 int64 sparse_index = shape_spec.processing_to_sparse_mapping[i]; in EmitDynamicSlice()
228 StridedSliceShapeSpec shape_spec; in Compile() local
236 &dummy, &dummy, &dummy, &begin, &end, &strides, &shape_spec)); in Compile()
292 int64 input_index = shape_spec.output_to_processing_mapping[i]; in Compile()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | training_ops_test.cc | 26 auto shape_spec = [&shape_spec_middle, shape_spec_end]( in TestGradAndIndicesErrorHandling() local
34 shape_spec("[?,1]", "[?,2];[?]")); in TestGradAndIndicesErrorHandling()
37 shape_spec("?", "[2,?];[1]")); in TestGradAndIndicesErrorHandling()
39 INFER_ERROR("must be equal rank", op, shape_spec("[1]", "[?,2];[?]")); in TestGradAndIndicesErrorHandling()
42 shape_spec("[?]", "[?];[1,2]")); in TestGradAndIndicesErrorHandling()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_tensor.cc | 106 const tstring& shape_spec = tensor_shapes_and_slices_ptr[i]; in SaveTensors() local
109 shape_spec, &shape, &slice, &slice_shape)); in SaveTensors()
115 shape_spec, ", tensor: ", input.shape().DebugString())); in SaveTensors()
195 const tstring& shape_spec = in RestoreTensor() local
197 if (!shape_spec.empty()) { in RestoreTensor()
200 shape_spec, &parsed_shape, &slice_to_load, in RestoreTensor()
|
| D | save_restore_v2_ops.cc | 118 const string& shape_spec = shape_and_slices_flat(i); in Compute() local
124 shape_spec, &shape, &slice, &slice_shape)); in Compute()
129 shape_spec, ", tensor: ", in Compute()
|