Home
last modified time | relevance | path

Searched refs:input0_shape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dset_ops.cc54 ShapeHandle input0_shape = c->input(0); in __anon1e42d5040102() local
55 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input0_shape, 2, &input0_shape)); in __anon1e42d5040102()
56 if (c->RankKnown(input0_shape)) { in __anon1e42d5040102()
57 const int32_t input0_rank = c->Rank(input0_shape); in __anon1e42d5040102()
66 c->Subshape(input0_shape, 0, rank - 1, &group0_shape)); in __anon1e42d5040102()
116 ShapeHandle input0_shape = c->input(0); in __anon1e42d5040202() local
117 TF_RETURN_IF_ERROR(c->WithRankAtLeast(input0_shape, 2, &input0_shape)); in __anon1e42d5040202()
118 if (c->RankKnown(input0_shape)) { in __anon1e42d5040202()
119 const int32_t input0_rank = c->Rank(input0_shape); in __anon1e42d5040202()
/external/tensorflow/tensorflow/lite/testing/op_tests/
Deinsum.py63 input0_shape = parameters["shapes"][0]
68 dtype=parameters["dtype"], shape=input0_shape)
76 input0_shape = set_dynamic_shape(parameters["shapes"][0])
78 input0_value = create_tensor_data(parameters["dtype"], input0_shape)
Dbatchmatmul.py80 input0_shape = parameters["shapes"][2]
85 input0_shape = swap_last_two_dims(*input0_shape)
87 parameters["dtype"], input0_shape, min_value=-1.0, max_value=1.0)
/external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/
Deinsum.py59 input0_shape = parameters["shapes"][0]
64 dtype=parameters["dtype"], shape=input0_shape)
72 input0_shape = parameters["shapes"][0]
74 input0_value = create_tensor_data(parameters["dtype"], input0_shape)
Dbatchmatmul.py82 input0_shape = parameters["shapes"][2]
87 input0_shape = swap_last_two_dims(*input0_shape)
89 parameters["dtype"], input0_shape, min_value=-1.0, max_value=1.0)
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_binary.cc99 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/ComputeLibrary/src/gpu/cl/operators/
DClWinogradConv2d.cpp135 …const TensorShape input0_shape = misc::shape_calculator::compute_winograd_input_transform_shape(*s… in validate_arguments() local
136 const TensorInfo input0 = src->clone()->set_tensor_shape(input0_shape); in validate_arguments()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h962 const RuntimeShape& input0_shape, const RuntimeShape& input1_shape, in NdArrayDescsForElementwiseBroadcast() argument
967 auto extended_input0_shape = RuntimeShape::ExtendedShape(N, input0_shape); in NdArrayDescsForElementwiseBroadcast()
995 const RuntimeShape& input0_shape, const RuntimeShape& input1_shape, in NdArrayDescsForElementwiseBroadcast() argument
1002 auto extended_input0_shape = RuntimeShape::ExtendedShape(N, input0_shape); in NdArrayDescsForElementwiseBroadcast()