| /external/tensorflow/tensorflow/core/ops/ |
| set_ops.cc | 58 ShapeHandle input1_shape = c->input(1);
60 c->WithRank(input1_shape, input0_rank, &input1_shape));
61 if (c->RankKnown(input1_shape)) {
63 const int32 rank = c->Rank(input1_shape);
69 c->Subshape(input1_shape, 0, rank - 1, &group1_shape));
76 ShapeHandle input1_shape = c->input(1);
77 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input1_shape, 2, &input1_shape));
78 if (c->RankKnown(input1_shape)) {
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| resolve_constant_binary.cc | 99 Shape input1_shape = input1_array.shape(); local
101 ExtendShape(&input1_shape, dims_count);
106 CHECK_EQ(input0_shape.dimensions_count(), input1_shape.dimensions_count());
109 VectorGreaterThan(input0_shape.dims(), input1_shape.dims());
113 PairwiseVectorSelect(input0_larger, input0_shape.dims(), input1_shape.dims(),
133 const auto val1 = input1_data[Offset(input1_shape, input1_indices)];
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_aggregate_ops.cc | 57 GetMklShape(ctx, src1_idx, &(mkl_context.input1_shape));
58 bool input1_in_mkl_format = mkl_context.input1_shape.IsMklTensor();
88 ? mkl_context.input1_shape.GetDimension()
111 ? &mkl_context.input1_shape
148 mkl_context.in_dims, mkl_context.input1_shape.GetTfToMklDimMap());
185 MklShape input1_shape, input2_shape, output_shape; member in struct:tensorflow::MklAddNOp::__anon39833
188 bool input1_in_mkl_format = input1_shape.IsMklTensor();
193 lt_input1 = static_cast<dnnLayout_t>(input1_shape.GetCurLayout());
271 bool input1_in_mkl_format = input1_shape.IsMklTensor();
|
