Home
last modified time | relevance | path

Searched refs:slice2 (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dslice_sinker_test.cc111 const HloInstruction* slice2; in TEST_F() local
116 m::Slice(&slice2, m::Add(m::Parameter(0), m::Parameter(1)))))); in TEST_F()
123 EXPECT_THAT(slice2->slice_starts(), ElementsAre(0, 0)); in TEST_F()
124 EXPECT_THAT(slice2->slice_limits(), ElementsAre(8, 4)); in TEST_F()
125 EXPECT_THAT(slice2->slice_strides(), ElementsAre(1, 1)); in TEST_F()
157 const HloInstruction* slice2; in TEST_F() local
164 m::Slice(&slice2, m::Multiply(m::Parameter(0), m::Parameter(1))), in TEST_F()
172 EXPECT_THAT(slice2->slice_starts(), ElementsAre(0, 0)); in TEST_F()
173 EXPECT_THAT(slice2->slice_limits(), ElementsAre(8, 2)); in TEST_F()
174 EXPECT_THAT(slice2->slice_strides(), ElementsAre(1, 1)); in TEST_F()
[all …]
Dfusion_node_indexing_evaluation_test.cc110 HloInstruction* slice2 = sub->mutable_operand(1); in TEST_F() local
112 EXPECT_EQ(instruction_fusion.EvaluateEmittedInstructions(slice2, fusion), 1); in TEST_F()
113 instruction_fusion.Fuse(slice2, fusion); in TEST_F()
Dalgebraic_simplifier_test.cc2356 HloInstruction* slice2 = builder.AddInstruction(HloInstruction::CreateSlice( in TEST_F() local
2392 {slice0, slice1, slice2, slice3, slice4, slice5, slice6, slice7, slice8}, in TEST_F()
/external/tensorflow/tensorflow/core/framework/
Dtensor_slice_test.cc295 TensorSlice slice2({{0, -1}}); in TEST() local
296 EXPECT_TRUE(slice2.IsFull()); in TEST()
305 TensorSlice slice2(2); in TEST() local
306 EXPECT_TRUE(slice1 != slice2); in TEST()
307 EXPECT_TRUE(slice2 != slice1); in TEST()
311 TensorSlice slice2({{0, -1}, {0, -1}, {0, -1}}); in TEST() local
312 EXPECT_TRUE(slice1 == slice2); in TEST()
313 EXPECT_TRUE(slice2 == slice1); in TEST()
317 TensorSlice slice2({{0, -1}, {0, 1}, {0, -1}}); in TEST() local
318 EXPECT_TRUE(slice1 != slice2); in TEST()
[all …]
/external/eigen/unsupported/test/
Dcxx11_tensor_morphing.cpp105 Tensor<float, 5, DataLayout> slice2(1,1,2,2,3); in test_simple_slice() local
108 slice2 = tensor.slice(indices2, sizes2); in test_simple_slice()
112 VERIFY_IS_EQUAL(slice2(0,0,i,j,k), tensor(1,1,3+i,4+j,5+k)); in test_simple_slice()
241 auto slice2 = SliceEvaluator(tensor.slice(offsets, extents), DefaultDevice()); in test_slice_raw_data() local
242 VERIFY_IS_EQUAL(slice2.dimensions().TotalSize(), 2); in test_slice_raw_data()
243 VERIFY_IS_EQUAL(slice2.data()[0], tensor(1,2,3,4)); in test_slice_raw_data()
244 VERIFY_IS_EQUAL(slice2.data()[1], tensor(2,2,3,4)); in test_slice_raw_data()
247 auto slice2 = SliceEvaluator(tensor.slice(offsets, extents), DefaultDevice()); in test_slice_raw_data() local
248 VERIFY_IS_EQUAL(slice2.dimensions().TotalSize(), 2); in test_slice_raw_data()
249 VERIFY_IS_EQUAL(slice2.data()[0], tensor(1,2,3,4)); in test_slice_raw_data()
[all …]
/external/grpc-grpc/test/core/compression/
Dstream_compression_test.cc202 grpc_slice slice2 = grpc_slice_buffer_take_first(&relay); in test_stream_compression_drop_context() local
204 grpc_slice_malloc(GRPC_SLICE_LENGTH(slice1) + GRPC_SLICE_LENGTH(slice2)); in test_stream_compression_drop_context()
208 GRPC_SLICE_START_PTR(slice2), GRPC_SLICE_LENGTH(slice2)); in test_stream_compression_drop_context()
210 grpc_slice_unref(slice2); in test_stream_compression_drop_context()
/external/webrtc/rtc_base/
Dcopy_on_write_buffer_unittest.cc354 CopyOnWriteBuffer slice2 = slice.Slice(2, 3); in TEST() local
355 EXPECT_EQ(slice2.size(), 3u); in TEST()
356 EXPECT_EQ(slice.cdata() + 2, slice2.cdata()); in TEST()
357 EXPECT_EQ(buf.cdata() + 5, slice2.cdata()); in TEST()
363 CopyOnWriteBuffer slice2 = buf.Slice(3, 7); in TEST() local
364 slice2[0] = 0xaa; in TEST()
/external/rust/crates/half/src/
Dslice.rs949 let slice2 = [0f32; 4]; in convert_from_f32_slice_len_mismatch_panics() localVariable
950 slice1.convert_from_f32_slice(&slice2); in convert_from_f32_slice_len_mismatch_panics()
957 let slice2 = [0f64; 4]; in convert_from_f64_slice_len_mismatch_panics() localVariable
958 slice1.convert_from_f64_slice(&slice2); in convert_from_f64_slice_len_mismatch_panics()
965 let mut slice2 = [0f32; 4]; in convert_to_f32_slice_len_mismatch_panics() localVariable
966 slice1.convert_to_f32_slice(&mut slice2); in convert_to_f32_slice_len_mismatch_panics()
973 let mut slice2 = [0f64; 4]; in convert_to_f64_slice_len_mismatch_panics() localVariable
974 slice1.convert_to_f64_slice(&mut slice2); in convert_to_f64_slice_len_mismatch_panics()
/external/perfetto/src/trace_processor/importers/common/
Dslice_tracker_unittest.cc346 SliceId slice2 = context.storage->slice_table().id()[1]; in TEST() local
348 EXPECT_EQ(tracker.GetTopmostSliceOnTrack(track).value(), slice2); in TEST()
385 SliceId slice2 = context.storage->slice_table().id()[1]; in TEST() local
387 EXPECT_THAT(slice_records, ElementsAre(slice1, slice2)); in TEST()
393 EXPECT_THAT(slice_records, ElementsAre(slice1, slice2, slice3)); in TEST()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dfused_slice.hlo73 // CHECK: slice2-after: ; preds = %[[VAL_70]], %[[VAL_66]]
85 // CHECK: slice2-true: ; preds = %[[VAL_66]]
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_writer_test.cc139 TensorSlice slice2(ss->slice()); in GetData() local
140 EXPECT_EQ(slice.DebugString(), slice2.DebugString()); in GetData()
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_math_ops.py935 slice2 = [slice(None)] * nd
937 slice2[axis] = slice(None, -1)
939 slice2 = tuple(slice2)
942 a = op(a[slice1], a[slice2])
/external/tensorflow/tensorflow/python/kernel_tests/
Dpartitioned_variables_test.py503 slice2 = _IotaInitializer([4, 5])
505 self.assertAllClose(slice0 + slice1 + slice2, val)
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle_test.cc567 TensorSlice slice2 = TensorSlice::ParseOrDie("-:1,9"); in TEST() local
573 TF_ASSERT_OK(writer.AddSlice("foo", kFullShape, slice2, in TEST()
604 EXPECT_EQ(slice2.DebugString(), slices[1].DebugString()); in TEST()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlower_static_tensor_list.cc292 TF::SliceOp slice2 = in matchAndRewrite() local
307 ArrayRef<Value>({slice1, expanded_item, slice2})); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/xla/tests/
Darray_elementwise_ops_test.cc412 auto slice2 = Slice(param, {1}, {slice_size + 1}, {1}); in XLA_TEST_F() local
413 return Add(slice1, slice2); in XLA_TEST_F()