Searched refs:input0_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | set_ops.cc | 54 ShapeHandle input0_shape = c->input(0); in __anonda6063e60102() local
55 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input0_shape, 2, &input0_shape)); in __anonda6063e60102()
56 if (c->RankKnown(input0_shape)) { in __anonda6063e60102()
57 const int32 input0_rank = c->Rank(input0_shape); in __anonda6063e60102()
66 c->Subshape(input0_shape, 0, rank - 1, &group0_shape)); in __anonda6063e60102()
116 ShapeHandle input0_shape = c->input(0); in __anonda6063e60202() local
117 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input0_shape, 2, &input0_shape)); in __anonda6063e60202()
118 if (c->RankKnown(input0_shape)) { in __anonda6063e60202()
119 const int32 input0_rank = c->Rank(input0_shape); in __anonda6063e60202()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_binary.cc | 99 Shape input0_shape = input0_array.shape(); in EvaluateBinaryOperatorOnConstantInputs() local
101 ExtendShape(&input0_shape, dims_count); in EvaluateBinaryOperatorOnConstantInputs()
107 CHECK_EQ(input0_shape.dimensions_count(), input1_shape.dimensions_count()); in EvaluateBinaryOperatorOnConstantInputs()
108 CHECK_EQ(input0_shape.dimensions_count(), dims_count); in EvaluateBinaryOperatorOnConstantInputs()
110 VectorGreaterThan(input0_shape.dims(), input1_shape.dims()); in EvaluateBinaryOperatorOnConstantInputs()
114 PairwiseVectorSelect(input0_larger, input0_shape.dims(), input1_shape.dims(), in EvaluateBinaryOperatorOnConstantInputs()
133 const auto val0 = input0_data[Offset(input0_shape, input0_indices)]; in EvaluateBinaryOperatorOnConstantInputs()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | common.h | 819 const RuntimeShape& input0_shape, const RuntimeShape& input1_shape, in NdArrayDescsForElementwiseBroadcast() argument
824 auto extended_input0_shape = RuntimeShape::ExtendedShape(N, input0_shape); in NdArrayDescsForElementwiseBroadcast()
852 const RuntimeShape& input0_shape, const RuntimeShape& input1_shape, in NdArrayDescsForElementwiseBroadcast() argument
859 auto extended_input0_shape = RuntimeShape::ExtendedShape(N, input0_shape); in NdArrayDescsForElementwiseBroadcast()
|