| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| bitcast_convert_test.cc | 127 // input -> reshape -> convert
129 // input -> convert -> reshape
134 auto reshape = Reshape(input, /*dimensions=*/{0}, /*new_sizes=*/{}); local
135 BitcastConvertType(reshape, F32);
|
| reshape_test.cc | 104 auto reshape = Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1}, local
106 auto new_shape = builder.GetShape(reshape).ConsumeValueOrDie();
121 Reshape(/*operand=*/a, /*dimensions=*/{}, /*new_sizes=*/{1});
201 Reshape(/*operand=*/parameter, /*dimensions=*/{0},
216 Reshape(/*operand=*/parameter, /*dimensions=*/{0},
231 Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1},
246 Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1},
263 Reshape(/*operand=*/parameter, /*dimensions=*/{1, 0},
309 Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1}
[all...] |
| select_and_scatter_test.cc | 279 const auto reshape = local
280 Reshape(operand, /*dimensions=*/{1, 0}, /*new_sizes=*/{2, 6});
283 SelectAndScatter(reshape, ge_s32_, /*window_dimensions=*/{2, 3},
|
| convert_test.cc | 417 // input -> reshape -> convert
419 // input -> convert -> reshape
424 auto reshape = Reshape(input, /*dimensions=*/{0}, /*new_sizes=*/{}); local
425 ConvertElementType(reshape, F32);
|
| multioutput_fusion_test.cc | 148 HloInstruction* reshape = local
155 ShapeUtil::MakeShapeWithDescendingLayout(F32, {1}), sub, reshape,
168 TF_CHECK_OK(reshape->ReplaceOperandWith(0, gte1));
|
| slice_test.cc | 174 auto reshape = Reshape(original, {24, 3, 2, 7}); local
175 Slice(reshape, {0, 0, 0, 0}, {11, 3, 2, 7}, {1, 1, 1, 1});
184 auto reshape = Reshape(original, {2 * 3 * 5, 7}); local
185 Slice(reshape, {0, 0}, {4, 7}, {1, 1});
|
| reduce_test.cc | 523 auto reshape = Reshape(log_, {rows, cols}); local
524 Reduce(reshape, zero, add_f32, /*dimensions_to_reduce=*/{0});
[all...] |
| fusion_test.cc | 336 auto reshape = builder.AddInstruction(HloInstruction::CreateReshape( local
339 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape},
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| const_analysis_test.cc | 40 auto c = ops::Reshape(root, arg2, b);
51 // Arg 1 must be constant because it flows to the shape argument of a Reshape.
52 // Arg 2 is used only as the value input to a Reshape and need not be const.
71 auto a = ops::Reshape(root, arg0, arg1);
72 auto b = ops::Reshape(root, arg2, a);
100 Output reshape = ops::Reshape(root, arg1, add); local
122 // Force const analysis to pretend that the shape argument to `reshape` does
124 Output reshape = ops::Reshape(root, arg1, add) local
[all...] |
| xla_compiler_test.cc | 345 auto reshape = ops::Reshape(scope, transposed_read, {2, 3}); local
346 auto d = ops::_Retval(scope.WithOpName("D"), reshape, 0);
428 auto c = ops::Reshape(scope.WithOpName("C"), a, b);
447 "reshape", std::move(graph), args,
[all...] |
| /external/tensorflow/tensorflow/contrib/metrics/python/kernel_tests/ |
| histogram_ops_test.py | 236 # Reshape [0, 1] to score_range.
237 def reshape(scores): function in function:synthetic_data
240 false_scores = reshape(false_scores)
241 true_scores = reshape(true_scores)
|
| /external/clang/test/Analysis/ |
| malloc-interprocedural.c | 77 static char *reshape(char *in) { function
83 v = reshape(v);
84 v = reshape(v);// expected-warning {{Potential leak of memory pointed to by 'v'}}
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| reshape.cc | 25 namespace reshape { namespace in namespace:tflite::ops::builtin
36 // Tensorflow's Reshape allows one of the shape components to have the
146 // reshape doesn't change the data, the output tensor needs exactly as many
159 } // namespace reshape
162 static TfLiteRegistration r = {nullptr, nullptr, reshape::Prepare,
163 reshape::Eval};
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| ops.py | 301 def reshape(labeled_tensor, existing_axes, new_axes, name=None): function
302 """Reshape specific axes of a LabeledTensor.
369 reshaped_tensor = array_ops.reshape(
399 return reshape(labeled_tensor, [existing_name], [new_axis], name=scope)
819 'Use transpose and reshape to create a single shared axis to sum '
853 a_tensor = array_ops.reshape(a.tensor, (1, -1))
861 b_tensor = array_ops.reshape(b.tensor, (-1, 1))
[all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| partially_decluster_pass_test.cc | 311 Output reshape = ops::Reshape(s.WithOpName("reshape"), reshape_input, shape); local
313 AddToCluster({shape.node(), reshape.node()}, "cluster_0");
338 Output reshape = ops::Reshape(s.WithOpName("reshape"), reshape_input, shape); local
340 AddToCluster({mul.node(), shape_of_mul.node(), shape.node(), reshape.node()},
363 Output reshape = ops::Reshape(s.WithOpName("reshape"), reshape_input, shape) local
388 Output reshape = ops::Reshape(s.WithOpName("reshape"), reshape_input, shape); local
427 Output reshape = local
[all...] |
| mark_for_compilation_pass_test.cc | 799 Output reshape = local
800 ops::Reshape(root.WithOpName("reshape"), reshape_input, shape);
843 Output reshape = local
871 Output reshape = local
902 Output reshape = local
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| tensor_array_ops.cc | 221 auto update = xla::Reshape(value, slice_shape.dim_sizes());
277 ctx->SetOutput(0, xla::Reshape(read, value_shape));
422 auto index = xla::Reshape(xla::Slice(indices, {i}, {i + 1}, {1}), {});
464 ctx->SetOutput(0, xla::Reshape(ta, shape));
533 const xla::XlaOp reshape = xla::Reshape(value, ta_shape.dim_sizes()); variable
535 ta = xla::Or(ta, reshape);
537 ta = xla::Add(ta, reshape);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| dynamic_dimension_inference_test.cc | 375 // Test the ability to trace unmodified reshape dimensions.
385 auto* reshape = builder.AddInstruction( local
400 EXPECT_EQ(inference_->GetDynamicSize(reshape, {}, 0), nullptr);
401 EXPECT_EQ(inference_->GetDynamicSize(reshape, {}, 1), size_param);
402 EXPECT_EQ(inference_->GetDynamicSize(reshape, {}, 2), nullptr);
403 EXPECT_EQ(inference_->GetDynamicSize(reshape, {}, 3), size_param);
404 EXPECT_EQ(inference_->GetDynamicSize(reshape, {}, 4), nullptr);
405 EXPECT_EQ(inference_->GetDynamicSize(reshape, {}, 5), nullptr);
409 // Test the ability to trace unmodified reshape dimensions.
|
| hlo_rematerialization_test.cc | 45 // F32[] %reshape = reshape(F32[], param)
63 auto reshape = builder.AddInstruction( local
66 HloInstruction::CreateBroadcast(vec1024_shape_, reshape, {}));
108 auto reshape = builder.AddInstruction( local
111 HloInstruction::CreateBroadcast(vec1024_shape_, reshape, {}));
|
| pattern_matcher_test.cc | 473 using match::Reshape;
482 r1 = u32[1] reshape(c1)
483 r2 = u32[1] reshape(c2)
484 r3 = u32[1] reshape(c3)
485 r4 = u32[1] reshape(c4)
492 Concatenate(Reshape(ConstantScalar(1)), Reshape(ConstantScalar(2)),
493 Reshape(ConstantScalar(3)), Reshape(ConstantScalar(4)))));
496 Concatenate(Reshape(ConstantScalar(2)), Reshape(ConstantScalar(1))
899 auto reshape = local
[all...] |
| hlo_computation_test.cc | 548 HloInstruction* reshape = local
557 HloInstruction::CreateDot(sout, x, reshape, dot_dnums, precision_config));
583 HloInstruction* reshape = local
592 HloInstruction::CreateDot(sout, x, reshape, dot_dnums, precision_config));
619 HloInstruction* reshape = local
628 HloInstruction::CreateDot(sout, x, reshape, dot_dnums, precision_config));
|
| layout_assignment_test.cc | 350 // param -> log -> reshape -> tanh
358 auto reshape = local
361 HloInstruction::CreateUnary(bshape, HloOpcode::kTanh, reshape));
383 AsInt64Slice(reshape->shape().layout().minor_to_major());
543 // param0 -> concatenate -> reshape
555 auto reshape = builder.AddInstruction( local
558 HloComputation* computation = m->AddEntryComputation(builder.Build(reshape));
[all...] |
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorBase.h | 738 reshape(const NewDimensions& newDimensions) const { function in class:Eigen::TensorBase
907 reshape(const NewDimensions& newDimensions) const { function in class:Eigen::TensorBase
912 reshape(const NewDimensions& newDimensions) { function in class:Eigen::TensorBase
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| xla.py | 395 def reshape(x, new_sizes, dimensions=None, name=None): function
398 x = array_ops.reshape(x, new_sizes, name=name)
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/ |
| __init__.py | 107 reshape = _ops.reshape variable
|
