| /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/lite/kernels/internal/reference/integer_ops/ |
| mul.h | 43 const RuntimeShape& input1_shape, const int8_t* input1_data,
50 MatchingFlatSize(input1_shape, input2_shape, output_shape);
57 const RuntimeShape& input1_shape, const int16* input1_data,
67 MatchingFlatSize(input1_shape, input2_shape, output_shape);
86 const RuntimeShape& input1_shape,
97 NdArrayDescsForElementwiseBroadcast(input1_shape, input2_shape, &desc1,
|
| add.h | 61 const RuntimeShape& input1_shape, const int8_t* input1_data,
67 MatchingFlatSize(input1_shape, input2_shape, output_shape);
78 const RuntimeShape& input1_shape,
87 NdArrayDescsForElementwiseBroadcast(input1_shape, input2_shape, &desc1,
|
| /external/tensorflow/tensorflow/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/lite/kernels/ |
| logical_test.cc | 28 LogicalOpModel(std::initializer_list<int> input1_shape,
34 BuildInterpreter({input1_shape, input2_shape});
|
| select_test.cc | 28 SelectOpModel(std::initializer_list<int> input1_shape,
38 BuildInterpreter({input1_shape, input2_shape, input3_shape});
|
| comparisons_test.cc | 28 ComparisonOpModel(std::initializer_list<int> input1_shape,
35 BuildInterpreter({input1_shape, input2_shape});
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| reference_ops.h | 526 const RuntimeShape& input1_shape, const T* input1_data,
530 MatchingFlatSize(input1_shape, input2_shape, output_shape);
539 const RuntimeShape& input1_shape, const float* input1_data,
542 const int size = MatchingFlatSize(input1_shape, input2_shape, output_shape);
634 const RuntimeShape& input1_shape, const uint8* input1_data,
640 MatchingFlatSize(input1_shape, input2_shape, output_shape);
650 const RuntimeShape& input1_shape, const int16* input1_data,
658 MatchingFlatSize(output_shape, input1_shape, input2_shape);
694 const RuntimeShape& input1_shape,
703 NdArrayDescsForElementwiseBroadcast(input1_shape, input2_shape, &desc1
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| common.h | 230 const RuntimeShape& input0_shape, const RuntimeShape& input1_shape,
236 auto extended_input1_shape = RuntimeShape::ExtendedShape(N, input1_shape);
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| optimized_ops.h | 6116 const RuntimeShape input1_shape = local
[all...] |
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| optimized_ops.h | 5788 const RuntimeShape input1_shape = local
[all...] |
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| common.h | 511 const RuntimeShape& input0_shape, const RuntimeShape& input1_shape,
517 auto extended_input1_shape = RuntimeShape::ExtendedShape(N, input1_shape);
|
