| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | convolution_test.cc | 442 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 …]
|
| D | prng_test.cc | 222 EXPECT_EQ(ShapeUtil::ElementsIn(actual.shape()), in XLA_TEST_F()
223 ShapeUtil::ElementsIn(param0_literal.shape())); in XLA_TEST_F()
224 for (int i = 0; i < ShapeUtil::ElementsIn(actual.shape()); ++i) { in XLA_TEST_F()
|
| D | reverse_test.cc | 83 ShapeUtil::ElementsIn(ShapeUtil::MakeShape(F32, spec.input_dims))); in TEST_P()
|
| /external/swiftshader/third_party/SPIRV-Tools/test/ |
| D | opcode_require_capabilities_test.cpp | 22 using spvtest::ElementsIn;
41 ElementsIn(GetParam().capabilities), in TEST_P()
42 ElementsIn(CapabilitySet(entry->numCapabilities, entry->capabilities))); in TEST_P()
|
| D | enum_set_test.cpp | 26 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()
|
| /external/deqp-deps/SPIRV-Tools/test/ |
| D | opcode_require_capabilities_test.cpp | 22 using spvtest::ElementsIn;
41 ElementsIn(GetParam().capabilities), in TEST_P()
42 ElementsIn(CapabilitySet(entry->numCapabilities, entry->capabilities))); in TEST_P()
|
| D | enum_set_test.cpp | 26 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()
|
| /external/angle/third_party/spirv-tools/src/test/ |
| D | opcode_require_capabilities_test.cpp | 22 using spvtest::ElementsIn;
41 ElementsIn(GetParam().capabilities), in TEST_P()
42 ElementsIn(CapabilitySet(entry->numCapabilities, entry->capabilities))); in TEST_P()
|
| D | enum_set_test.cpp | 26 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()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis.cc | 98 auto computation_count = ShapeUtil::ElementsIn(shape); in HandleElementwiseOp()
331 kFmaFlops * ShapeUtil::ElementsIn(dot_shape) * reduction_width; in HandleDot()
349 const int64 element_count = ShapeUtil::ElementsIn(map->shape()); in HandleMap()
373 ShapeUtil::ElementsIn(arg->shape()) - ShapeUtil::ElementsIn(output_shape); in HandleReduce()
397 ShapeUtil::ElementsIn(reduce_window->shape()); in HandleReduceWindow()
421 const auto source_element_count = ShapeUtil::ElementsIn(source->shape()); in HandleSelectAndScatter()
668 ShapeUtil::ElementsIn(real_shape); in HandleFft()
686 elems *= ShapeUtil::ElementsIn(b_shape); in HandleTriangularSolve()
702 elems *= ShapeUtil::ElementsIn(a_shape); in HandleCholesky()
717 flops += ShapeUtil::ElementsIn(subshape); in HandleAllReduce()
[all …]
|
| D | hlo_constant_folding.cc | 127 ShapeUtil::ElementsIn(operand->shape()); in Run()
131 ShapeUtil::ElementsIn(instruction->shape()); in Run()
|
| D | slice_sinker.cc | 132 sum += ShapeUtil::ElementsIn(user->shape()); in ShouldTransform()
134 return sum >= xla::ShapeUtil::ElementsIn( in ShouldTransform()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 352 TEST(ShapeUtilTest, ElementsIn) { in TEST() argument
353 EXPECT_EQ(1, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {}))); in TEST()
354 EXPECT_EQ(0, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {0}))); in TEST()
355 EXPECT_EQ(1, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {1}))); in TEST()
356 EXPECT_EQ(1, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {1, 1}))); in TEST()
357 EXPECT_EQ(2, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {2}))); in TEST()
358 EXPECT_EQ(2, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {2, 1}))); in TEST()
359 EXPECT_EQ(15, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {3, 5}))); in TEST()
360 EXPECT_EQ(0, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {3, 0, 5}))); in TEST()
361 EXPECT_EQ(0, ShapeUtil::ElementsIn(ShapeUtil::MakeShape(S32, {0, 3, 0}))); in TEST()
[all …]
|
| D | packed_literal_reader.cc | 62 int64 elements = ShapeUtil::ElementsIn(shape); in Read()
|
| D | literal.h | 232 return ShapeUtil::ElementsIn(shape());
234 return ShapeUtil::ElementsIn(ShapeUtil::GetSubshape(shape(), index));
366 int64 element_count() const { return ShapeUtil::ElementsIn(subshape()); } in element_count()
928 CHECK_EQ(ShapeUtil::ElementsIn(shape()), values.size()); in PopulateR1()
1061 int64 elements = ShapeUtil::ElementsIn(literal.shape()); in Replicate()
|
| D | shape_util.cc | 425 /* static */ int64 ShapeUtil::ElementsIn(const Shape& shape) { in ElementsIn() function in xla::ShapeUtil
439 return ElementsIn(shape); in ElementsInRecursive()
462 return shape.IsArray() && ElementsIn(shape) == 0; in IsZeroElementArray()
655 allocated_element_count = ElementsIn(shape); in ByteSizeOfElements()
1138 CHECK_EQ(ElementsIn(input_shape), ElementsIn(output_shape)); in ReshapeIsBitcast()
1139 if (ElementsIn(input_shape) == 0) { in ReshapeIsBitcast()
|
| D | literal_util.cc | 351 CHECK_EQ(ShapeUtil::ElementsIn(literal.shape()), new_num_elements); in ReshapeSlice()
364 for (int64 i = 0; i < ShapeUtil::ElementsIn(literal.shape()); ++i) { in ReshapeSlice()
422 CHECK_GT(ShapeUtil::ElementsIn(literal.shape()), 0); in GetFirstScalarLiteral()
|
| D | index_util.cc | 95 DCHECK_LT(linear_index, ShapeUtil::ElementsIn(shape)); in LinearIndexToMultidimensionalIndex()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 71 int64 elements_before = xla::ShapeUtil::ElementsIn(input.shape()); in ReshapeLiteral()
72 int64 elements_after = xla::ShapeUtil::ElementsIn(shape); in ReshapeLiteral()
|
| D | literal_util.cc | 81 xla::ShapeUtil::ElementsIn(literal.shape()) == in CopyLiteralToHostTensor()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_rewriter.cc | 77 if (ShapeUtil::ElementsIn(batch_norm->operand(0)->shape()) == 0) { in HandleBatchNormInference()
129 if (ShapeUtil::ElementsIn(batch_norm->operand(0)->shape()) == 0) { in HandleBatchNormTraining()
218 if (ShapeUtil::ElementsIn(batch_norm->operand(0)->shape()) == 0) { in HandleBatchNormGrad()
|
| D | fft_thunk.cc | 178 .ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
202 .ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
|
| D | partition_assignment.cc | 65 int64 num_elements = ShapeUtil::ElementsIn(shape); in CalculateLaunchDimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 152 const int64 size = ShapeUtil::ElementsIn(shape); in ThreeFryRngBit32()
170 const int64 size = ShapeUtil::ElementsIn(shape); in ThreeFryRngBit64()
334 const int64 num_elems = ShapeUtil::ElementsIn(shape); in PhiloxRngBit32()
362 const int64 num_elems = ShapeUtil::ElementsIn(shape); in PhiloxRngBit64()
524 const int64 num_elems = ShapeUtil::ElementsIn(shape); in NormalFloatingPointDistribution()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emission_utils.cc | 37 ShapeUtil::ElementsIn(shape) * in GetMinimumAlignmentForArray()
|