Home
last modified time | relevance | path

Searched refs:ElementsIn (Results 1 – 25 of 49) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc442 std::vector<float> input_elems(ShapeUtil::ElementsIn(input_shape)); in XLA_TEST_F()
447 std::vector<float> filter_elems(ShapeUtil::ElementsIn(filter_shape)); in XLA_TEST_F()
505 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape)); in RunTest()
510 std::vector<T> filter_elems(ShapeUtil::ElementsIn(filter_shape)); in RunTest()
565 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape)); in RunTest()
570 std::vector<T> filter_elems(ShapeUtil::ElementsIn(filter_shape)); in RunTest()
631 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape)); in RunTest()
636 std::vector<T> filter_elems(ShapeUtil::ElementsIn(filter_shape)); in RunTest()
701 std::vector<T> input_elems(ShapeUtil::ElementsIn(input_shape), in RunTest()
706 std::vector<T> filter_elems(ShapeUtil::ElementsIn(filter_shape), in RunTest()
[all …]
Dprng_test.cc214 EXPECT_EQ(ShapeUtil::ElementsIn(actual.shape()), in XLA_TEST_F()
215 ShapeUtil::ElementsIn(param0_literal.shape())); in XLA_TEST_F()
216 for (int i = 0; i < ShapeUtil::ElementsIn(actual.shape()); ++i) { in XLA_TEST_F()
Dreverse_test.cc83 ShapeUtil::ElementsIn(ShapeUtil::MakeShape(F32, spec.input_dims))); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis.cc90 auto computation_count = ShapeUtil::ElementsIn(shape); in HandleElementwiseOp()
251 kFmaFlops * ShapeUtil::ElementsIn(dot_shape) * reduction_width; in HandleDot()
269 const int64 element_count = ShapeUtil::ElementsIn(map->shape()); in HandleMap()
293 ShapeUtil::ElementsIn(arg->shape()) - ShapeUtil::ElementsIn(output_shape); in HandleReduce()
317 ShapeUtil::ElementsIn(reduce_window->shape()); in HandleReduceWindow()
341 const auto source_element_count = ShapeUtil::ElementsIn(source->shape()); in HandleSelectAndScatter()
546 ShapeUtil::ElementsIn(real_shape); in HandleFft()
559 elems *= ShapeUtil::ElementsIn(b_shape); in HandleTriangularSolve()
571 elems *= ShapeUtil::ElementsIn(a_shape); in HandleCholesky()
586 flops += ShapeUtil::ElementsIn(subshape); in HandleAllReduce()
[all …]
Dhlo_constant_folding.cc127 ShapeUtil::ElementsIn(operand->shape()); in Run()
131 ShapeUtil::ElementsIn(instruction->shape()); in Run()
/external/swiftshader/third_party/SPIRV-Tools/test/
Denum_set_test.cpp26 using spvtest::ElementsIn;
241 EXPECT_THAT(ElementsIn(GetParam().capabilities), Eq(GetParam().expected)); in TEST_P()
246 EXPECT_THAT(ElementsIn(copy), Eq(GetParam().expected)); in TEST_P()
253 EXPECT_THAT(ElementsIn(moved), Eq(GetParam().expected)); in TEST_P()
256 EXPECT_THAT(ElementsIn(copy), Eq(std::vector<SpvCapability>{})); in TEST_P()
261 EXPECT_THAT(ElementsIn(assigned), Eq(GetParam().expected)); in TEST_P()
267 EXPECT_THAT(ElementsIn(assigned), Eq(GetParam().expected)); in TEST_P()
Dopcode_require_capabilities_test.cpp22 using spvtest::ElementsIn;
41 ElementsIn(GetParam().capabilities), in TEST_P()
42 ElementsIn(CapabilitySet(entry->numCapabilities, entry->capabilities))); in TEST_P()
/external/deqp-deps/SPIRV-Tools/test/
Denum_set_test.cpp26 using spvtest::ElementsIn;
241 EXPECT_THAT(ElementsIn(GetParam().capabilities), Eq(GetParam().expected)); in TEST_P()
246 EXPECT_THAT(ElementsIn(copy), Eq(GetParam().expected)); in TEST_P()
253 EXPECT_THAT(ElementsIn(moved), Eq(GetParam().expected)); in TEST_P()
256 EXPECT_THAT(ElementsIn(copy), Eq(std::vector<SpvCapability>{})); in TEST_P()
261 EXPECT_THAT(ElementsIn(assigned), Eq(GetParam().expected)); in TEST_P()
267 EXPECT_THAT(ElementsIn(assigned), Eq(GetParam().expected)); in TEST_P()
Dopcode_require_capabilities_test.cpp22 using spvtest::ElementsIn;
41 ElementsIn(GetParam().capabilities), in TEST_P()
42 ElementsIn(CapabilitySet(entry->numCapabilities, entry->capabilities))); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc330 TEST(ShapeUtilTest, ElementsIn) { in TEST() argument
331 EXPECT_EQ(1, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {}))); in TEST()
332 EXPECT_EQ(0, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {0}))); in TEST()
333 EXPECT_EQ(1, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {1}))); in TEST()
334 EXPECT_EQ(1, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {1, 1}))); in TEST()
335 EXPECT_EQ(2, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {2}))); in TEST()
336 EXPECT_EQ(2, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {2, 1}))); in TEST()
337 EXPECT_EQ(15, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {3, 5}))); in TEST()
338 EXPECT_EQ(0, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {3, 0, 5}))); in TEST()
339 EXPECT_EQ(0, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {0, 3, 0}))); in TEST()
[all …]
Dshape_util.cc378 /* static */ int64 ShapeUtil::ElementsIn(const Shape& shape) { in ElementsIn() function in xla::ShapeUtil
392 return ElementsIn(shape); in ElementsInRecursive()
415 return shape.IsArray() && ElementsIn(shape) == 0; in IsZeroElementArray()
608 allocated_element_count = ElementsIn(shape); in ByteSizeOfElements()
1111 CHECK_EQ(ElementsIn(input_shape), ElementsIn(output_shape)); in ReshapeIsBitcast()
1112 if (ElementsIn(input_shape) == 0) { in ReshapeIsBitcast()
Dliteral_util.cc299 CHECK_EQ(ShapeUtil::ElementsIn(literal.shape()), new_num_elements); in ReshapeSlice()
312 for (int64 i = 0; i < ShapeUtil::ElementsIn(literal.shape()); ++i) { in ReshapeSlice()
370 CHECK_GT(ShapeUtil::ElementsIn(literal.shape()), 0); in GetFirstScalarLiteral()
Dpacked_literal_reader.cc62 int64 elements = ShapeUtil::ElementsIn(shape); in Read()
Dliteral.h219 return ShapeUtil::ElementsIn(shape());
221 return ShapeUtil::ElementsIn(ShapeUtil::GetSubshape(shape(), index));
372 : ShapeUtil::ElementsIn(subshape()); in element_count()
994 CHECK_EQ(ShapeUtil::ElementsIn(shape()), values.size()); in PopulateR1()
1152 int64 elements = ShapeUtil::ElementsIn(literal.shape()); in Replicate()
Dindex_util.cc95 DCHECK_LT(linear_index, ShapeUtil::ElementsIn(shape)); in LinearIndexToMultidimensionalIndex()
Dliteral_comparison.cc197 return ShapeUtil::ElementsIn(shape); in RecursiveElementCount()
544 int64 element_count = ShapeUtil::ElementsIn(actual_.shape()); in ErrorMessage()
557 ShapeUtil::ElementsIn(actual_.shape()), error_.abs, error_.rel); in ErrorMessage()
781 ShapeUtil::ElementsIn(expected.shape()) >= 64); in NearHelper()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc62 if (ShapeUtil::ElementsIn(batch_norm->operand(0)->shape()) == 0) { in HandleBatchNormInference()
98 if (ShapeUtil::ElementsIn(batch_norm->operand(0)->shape()) == 0) { in HandleBatchNormTraining()
167 if (ShapeUtil::ElementsIn(batch_norm->operand(0)->shape()) == 0) { in HandleBatchNormGrad()
Dfft_thunk.cc178 ->ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
202 ->ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
Dpartition_assignment.cc65 int64 num_elements = ShapeUtil::ElementsIn(shape); in CalculateLaunchDimensions()
Dparallel_loop_emitter.cc135 llvm::ConstantInt::get(index_type, ShapeUtil::ElementsIn(shape_))), in EmitIndexAndSetExitBasicBlock()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.cc71 int64 elements_before = xla::ShapeUtil::ElementsIn(input.shape()); in ReshapeLiteral()
72 int64 elements_after = xla::ShapeUtil::ElementsIn(shape); in ReshapeLiteral()
Dliteral_util.cc81 xla::ShapeUtil::ElementsIn(literal.shape()) == in CopyLiteralToHostTensor()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dprng.cc117 const int64 size = ShapeUtil::ElementsIn(shape); in StatelessRngUniformU32()
146 const int64 size = ShapeUtil::ElementsIn(shape); in StatelessRngUniformU64()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc37 ShapeUtil::ElementsIn(shape) * in GetMinimumAlignmentForArray()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstateful_random_ops.cc54 const int64 size = xla::ShapeUtil::ElementsIn(shape); in StatefulRngUniformU32()
71 const int64 size = xla::ShapeUtil::ElementsIn(shape); in StatefulRngUniformU64()

12