| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| dynamic_stitch_op.cc | 141 TensorShape new_shape; variable
143 new_shape.AddDim(indices[input_num].shape().dimensions(0));
146 new_shape.AddDim(data0_shape.dim_size(d));
150 if (new_shape == data_shapes[input_num]) {
154 ctx->builder()->Reshape(handle, new_shape.dim_sizes());
|
| shape_op.cc | 135 std::vector<int64> new_shape(existing_dims_size);
136 for (size_t i = 0; i < new_shape.size(); ++i) {
137 new_shape[i] = existing_dims[i];
148 new_shape.emplace(new_shape.begin() + dim, 1);
150 ctx->SetOutput(0, ctx->builder()->Reshape(ctx->Input(0), new_shape));
167 std::vector<int64> new_shape; variable
198 new_shape.push_back(existing_dim);
203 new_shape.push_back(existing_dim);
208 ctx->SetOutput(0, ctx->builder()->Reshape(ctx->Input(0), new_shape));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_element_type_converter.cc | 186 Shape new_shape = GetConvertedTupleShape(hlo->shape(), eliminate_type_, local
189 new_shape, new_operands, hlo->GetModule()));
|
| reshape_mover.cc | 196 const Shape new_shape = local
204 HloInstruction::CreateReshape(new_shape, operand));
211 new_shape, operand, inverse_permutation));
218 operand->CloneWithNewOperands(new_shape, operand->operands()));
|
| layout_assignment.cc | 515 Shape new_shape = channel_constraints->LayoutShapeForChannel( local
518 constraints->SetBufferLayout(new_shape.layout(), *buffer));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| reshape_test.cc | 106 auto new_shape = builder.GetShape(reshape).ConsumeValueOrDie(); local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| shape_ops.h | 167 std::vector<int64> new_shape(existing_dims_size);
168 for (size_t i = 0; i < new_shape.size(); ++i) {
169 new_shape[i] = existing_dims[i];
180 new_shape.emplace(new_shape.begin() + dim, 1);
181 const TensorShape output_shape(new_shape);
212 std::vector<int64> new_shape; variable
244 new_shape.push_back(existing_dim);
249 new_shape.push_back(existing_dim);
254 const TensorShape output_shape(new_shape);
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| array_ops_test.cc | 825 Tensor new_shape = test::AsTensor<int32>({1, 2, 3}); local
826 op.input_tensors[1] = &new_shape;
836 new_shape = test::AsTensor<int32>({-1});
840 new_shape = test::AsTensor<int32>({2, -1});
847 new_shape = test::AsTensor<int32>({-1, -1, 2});
851 new_shape = test::AsTensor<int32>({});
858 new_shape = test::AsTensor<int32>({-1});
860 new_shape = test::AsTensor<int32>({-1, 6});
862 new_shape = test::AsTensor<int32>({0, -1});
876 Tensor new_shape = test::AsTensor<int32>({1, 2, 3}) local
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/ |
| model.cc | 485 auto* new_shape = schema_params->new_shape(); local
486 FlatBufferIntVectorToArray(sizeof(params->shape), new_shape,
488 params->num_dimensions = new_shape->Length();
|
| /external/tensorflow/tensorflow/core/framework/ |
| shape_inference.h | 247 ShapeHandle new_shape; local
248 if (!Merge(inputs_[idx], shape, &new_shape).ok()) return false;
249 inputs_[idx] = new_shape;
279 ShapeHandle new_shape; local
280 Relax(inputs_[idx], shape, &new_shape);
281 if (inputs_[idx].SameHandle(new_shape)) {
284 inputs_[idx] = new_shape;
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| single_machine_test.cc | 163 auto new_shape = ops::Const(root.WithOpName("new_shape"), {3, -1}, {2}); local
164 auto reshaped = ops::Reshape(root.WithOpName("reshaped"), square, new_shape);
206 "new_shape", "reshaped", "final_shape", "expected_shape",
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| literal_util.cc | 708 Shape new_shape = shape(); local
709 Shape* subshape = ShapeUtil::GetMutableSubshape(&new_shape, shape_index);
712 auto result = MakeUnique<Literal>(new_shape);
[all...] |
| shape_util.cc | 811 Shape new_shape = original; local
812 new_shape.set_element_type(type);
813 return new_shape;
973 Shape new_shape = shape; local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_parser.cc | 1361 Shape new_shape; local
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| constant_folding.cc | 1234 const NodeDef* new_shape = node_map_->GetNode(node.input(1)); local
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api.cc | 1020 tensorflow::shape_inference::ShapeHandle new_shape = local
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/schema/ |
| schema_generated.h | 2691 std::vector<int32_t> new_shape; member in struct:tflite::ReshapeOptionsT
2701 const flatbuffers::Vector<int32_t> *new_shape() const { function in struct:tflite::FLATBUFFERS_FINAL_CLASS
[all...] |
