| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| aggregate_ops.cc | 47 TensorShape operand_shape; variable
49 ctx, GetTensorListBufferShape(ctx->Input(i), &operand_shape));
51 ctx, sum_shape.dim_sizes() == operand_shape.dim_sizes(),
55 "Found: ", operand_shape.DebugString()));
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| gather_operation_test.cc | 641 Shape operand_shape = ShapeUtil::MakeShape(S32, {3, 3}); local
644 auto operand = Parameter(&builder, 0, operand_shape, "operand");
|
| select_and_scatter_test.cc | 42 std::vector<int64> operand_shape; member in struct:xla::__anon44659::SelectAndScatterTestParam
73 auto operand_shape = GetParam().operand_shape; local
74 Array<float> o(operand_shape);
|
| test_utils.cc | 469 const Shape& operand_shape = use->operand(0)->shape(); local
478 std::min(index_bound, operand_shape.dimensions(dim_in_operand));
|
| reduce_test.cc | 939 Shape operand_shape = ShapeUtil::MakeShape(F32, {1}); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| pooling_test.cc | 39 Shape operand_shape = b->GetShape(input).ValueOrDie(); local
40 std::vector<int64> input_size(operand_shape.dimensions().begin(),
41 operand_shape.dimensions().end());
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| fuse_binary_into_following_affine.cc | 202 const auto& operand_shape = local
204 for (const auto& dim : operand_shape.dims()) {
|
| fuse_binary_into_preceding_affine.cc | 45 const Shape& operand_shape = operand.shape(); local
54 if (operand_shape.dimensions_count() >= 1 &&
55 operand_shape.dims(operand_shape.dimensions_count() - 1) ==
58 } else if (operand_shape.dimensions_count() == 0 ||
59 operand_shape.dims(operand_shape.dimensions_count() - 1) == 1) {
114 const Shape& operand_shape = operand.shape(); local
125 if (operand_shape.dimensions_count() >= 1 &&
126 operand_shape.dims(operand_shape.dimensions_count() - 1) =
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_cost_analysis_test.cc | 645 Shape operand_shape = ShapeUtil::MakeShape(S32, {3, 3}); local
648 auto operand = Parameter(&builder, 0, operand_shape, "operand");
670 Shape operand_shape = ShapeUtil::MakeShape(F32, {3, 3}); local
674 auto operand = Parameter(&builder, 0, operand_shape, "operand");
|
| hlo_creation_utils.cc | 314 const Shape& operand_shape = operand->shape(); local
315 CHECK_GE(operand_shape.dimensions_size(), n);
318 new_shape_leading_bound *= operand_shape.dimensions(i);
322 new_shape_dims.reserve(operand_shape.dimensions_size() - n + 1);
325 std::copy(operand_shape.dimensions().begin() + n,
326 operand_shape.dimensions().end(),
330 ShapeUtil::MakeShape(operand_shape.element_type(), new_shape_dims);
339 const Shape& operand_shape = operand->shape(); local
340 new_shape_dims.reserve(n + operand_shape.dimensions_size());
342 absl::c_copy(operand_shape.dimensions(), std::back_inserter(new_shape_dims))
392 const Shape& operand_shape = operand->shape(); local
[all...] |
| batchnorm_expander.cc | 206 const Shape operand_shape = operand->shape(); local
207 PrimitiveType ptype = operand_shape.element_type();
221 operand_shape,
225 for (int64 i = 0; i < operand_shape.rank(); ++i) {
235 HloInstruction::CreateBroadcast(operand_shape, scale, {feature_index}));
238 HloInstruction::CreateBroadcast(operand_shape, offset, {feature_index}));
245 add_binary(operand_shape, HloOpcode::kMultiply, operand, operand);
260 HloInstruction::CreateBroadcast(operand_shape, mean, {feature_index}));
274 add(HloInstruction::CreateBroadcast(operand_shape, var, {feature_index}));
278 add_binary(operand_shape, HloOpcode::kAdd, var_broadcasted, epsilon)
333 const Shape operand_shape = operand->shape(); local
[all...] |
| hlo_evaluator.cc | 465 const Shape& operand_shape = operands[i]->shape(); local
466 CHECK(operand_shape.IsArray());
470 ShapeUtil::GetDimension(operand_shape, concat_dim);
479 const Shape& operand_shape = operand->shape(); local
482 AsInt64Slice(operand_shape.dimensions())));
484 ShapeUtil::GetDimension(operand_shape, concat_dim);
1083 const Shape& operand_shape = operand.shape(); local
[all...] |
| elemental_ir_emitter.cc | 1863 const Shape& operand_shape = hlo->operand(0)->shape(); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| ir_emitter.cc | 482 const Shape& operand_shape = operand->shape(); local
485 if (!operand_shape.IsTuple()) {
486 return EmitXfeedTransfer(XfeedKind::kOutfeed, operand_shape, value);
489 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(operand_shape));
491 for (int64 i = 0; i < operand_shape.tuple_shapes_size(); ++i) {
493 ShapeUtil::GetTupleElementShape(operand_shape, i);
1337 const Shape& operand_shape = crs->operand(i)->shape(); local
1370 const Shape& operand_shape = reduce.operand(0)->shape(); local
[all...] |
