Searched refs:dims_tensor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unravel_index_op.cc | 49 const Tensor& dims_tensor = ctx->input(1); in Compute() local
51 ctx, TensorShapeUtils::IsVector(dims_tensor.shape()), in Compute()
53 dims_tensor.shape().DebugString(), "\"")); in Compute()
55 auto dims = dims_tensor.vec<Tidx>(); in Compute()
72 TensorShape({dims_tensor.NumElements()}), in Compute()
83 TensorShape({dims_tensor.NumElements()}), in Compute()
94 ctx, ctx->allocate_output(0, TensorShape({dims_tensor.NumElements()}), in Compute()
104 TensorShape({dims_tensor.NumElements(), in Compute()
111 {static_cast<Eigen::Index>(dims_tensor.NumElements()), 1}}; in Compute()
117 {static_cast<Eigen::Index>(dims_tensor.NumElements()), 1}); in Compute()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 2720 const auto& dims_tensor = context->tensors[node->inputs->data[0]]; in Validate() local
2721 Expect(IsConstantTensor(&dims_tensor), in Validate()
2724 EXPECT_INPUT_TYPE_IN(dims_tensor.type, kTfLiteInt32, kTfLiteInt64); in Validate()
2725 if (IsConstantTensor(&dims_tensor)) { in Validate()
2726 Expect(dims_tensor.dims->data[0] != 0, in Validate()
2729 if (dims_tensor.type == kTfLiteInt64) { in Validate()
2731 std::all_of(dims_tensor.data.i64, in Validate()
2732 dims_tensor.data.i64 + dims_tensor.dims->data[0], in Validate()
4591 const TfLiteTensor& dims_tensor = context->tensors[dims_id]; in AddOpsAndTensors() local
4592 switch (dims_tensor.type) { in AddOpsAndTensors()
[all …]
|