Home
last modified time | relevance | path

Searched refs:flat_outer_dims (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtensor_test.cc547 auto flat_outer_dims = t.flat_outer_dims<float>(); in TEST_F() local
548 EXPECT_EQ(2, flat_outer_dims.dimension(0)); in TEST_F()
549 EXPECT_EQ(60, flat_outer_dims.dimension(1)); in TEST_F()
550 EXPECT_EQ(flat_outer_dims(0, 0), 0.01f); in TEST_F()
551 EXPECT_EQ(flat_outer_dims(1, 59), 0.02f); in TEST_F()
554 auto flat_outer_dims = t.flat_outer_dims<float, 3>(); in TEST_F() local
555 EXPECT_EQ(2, flat_outer_dims.dimension(0)); in TEST_F()
556 EXPECT_EQ(3, flat_outer_dims.dimension(1)); in TEST_F()
557 EXPECT_EQ(20, flat_outer_dims.dimension(2)); in TEST_F()
558 EXPECT_EQ(flat_outer_dims(0, 0, 0), 0.01f); in TEST_F()
[all …]
Dtensor.h323 typename TTypes<T, NDIMS>::Tensor flat_outer_dims();
424 typename TTypes<T, NDIMS>::ConstTensor flat_outer_dims() const;
733 typename TTypes<T, NDIMS>::Tensor Tensor::flat_outer_dims() { in flat_outer_dims() function
750 typename TTypes<T, NDIMS>::ConstTensor Tensor::flat_outer_dims() const { in flat_outer_dims() function
/external/tensorflow/tensorflow/core/kernels/
Dbatch_util.cc47 parent->flat_outer_dims<T>().chip(index, 0) = element.flat<T>(); in HandleElementToSlice()
54 auto parent_as_matrix = parent->flat_outer_dims<string>(); in HandleElementToSlice()
69 auto parent_as_matrix = parent->flat_outer_dims<Variant>(); in HandleElementToSlice()
87 element->flat<T>() = parent.flat_outer_dims<T>().chip(index, 0); in HandleSliceToElement()
Dcwise_op_select.cc79 FastBoundsCheck(then->flat_outer_dims<T>().dimension(1), in ComputeBroadcasting()
105 func(ctx->eigen_device<Device>(), output->flat_outer_dims<T>(), in ComputeBroadcasting()
106 cond->vec<bool>(), then->flat_outer_dims<T>(), in ComputeBroadcasting()
107 else_->flat_outer_dims<T>()); in ComputeBroadcasting()
Dtraining_ops.cc775 auto var_flat = var.flat_outer_dims<T>(); in DoCompute()
776 auto accum_grad_flat = accum_grad.flat_outer_dims<T>(); in DoCompute()
777 auto accum_update_flat = accum_update.flat_outer_dims<T>(); in DoCompute()
778 auto grad_flat = grad.flat_outer_dims<T>(); in DoCompute()
958 auto var_flat = var.flat_outer_dims<T>(); in Compute()
959 auto grad_flat = grad.flat_outer_dims<T>(); in Compute()
1299 auto var_flat = var.flat_outer_dims<T>(); in Compute()
1300 auto accum_flat = accum.flat_outer_dims<T>(); in Compute()
1301 auto grad_flat = grad.flat_outer_dims<T>(); in Compute()
1450 auto var_flat = var.flat_outer_dims<T>(); in Compute()
[all …]
Drandom_shuffle_op.cc93 const auto input_mat = input.flat_outer_dims<T>(); in Compute()
94 auto output_mat = output->flat_outer_dims<T>(); in Compute()
Dsegment_reduction_ops.cc89 auto input_flat = input.flat_outer_dims<T>(); in Compute()
112 auto output_flat = output->flat_outer_dims<T>(); in Compute()
282 auto output_flat = output->flat_outer_dims<T>(); in ComputeAsync()
489 auto output_flat = output->flat_outer_dims<T>(); in Compute()
654 auto input_flat = input.flat_outer_dims<T>(); in Compute()
685 auto output_flat = output->flat_outer_dims<T>(); in Compute()
1023 auto input_flat = input.flat_outer_dims<T>(); in Compute()
1060 auto output_flat = output->flat_outer_dims<T>(); in Compute()
Dsparse_conditional_accumulator.h264 auto sum_flat = sum_tensor->flat_outer_dims<T>(); in AddToAccumGradFunction()
265 auto accum_flat = accum_val_->flat_outer_dims<T>(); in AddToAccumGradFunction()
266 auto grad_flat = grad_val->flat_outer_dims<T>(); in AddToAccumGradFunction()
324 auto accum_flat = accum_val_->flat_outer_dims<T>(); in DivideAccumGradByCounter()
Dscatter_op.cc124 auto params_flat = params.flat_outer_dims<T>(); in DoCompute()
197 auto params_flat = params.flat_outer_dims<T>(); in DoCompute()
Dinplace_ops_functor_gpu.cu.cc47 auto Toutput = output->flat_outer_dims<T>(); in DoParallelConcatUpdate()
Ddynamic_stitch_op.cc175 const int slice_size = merged->flat_outer_dims<T>().dimension(1); in Compute()
239 auto merged_flat = merged->flat_outer_dims<T>(); in Compute()
Ddynamic_partition_op.cc132 out_flat.push_back(outputs[p]->flat_outer_dims<T>()); in Compute()
Dinplace_ops.cc38 auto Toutput = output->flat_outer_dims<T>(); in DoParallelConcatUpdate()
Dgather_nd_op.cc169 auto params_flat = params.flat_outer_dims<T, IXDIM + 1>(); \ in DoGatherNd()
Dqueue_base.cc45 auto parent_as_matrix = parent.flat_outer_dims<T>(); in HandleSliceToElement()
Dbarrier_ops.cc313 auto values_matrix = values.flat_outer_dims<T>(); in InsertOneLocked()
Dresource_variable_ops.cc599 auto params_flat = params->flat_outer_dims<T>(); in Compute()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dstats_accumulator_ops.cc196 auto gradients = gradients_t->flat_outer_dims<float>(); in SerializeTensorAccumulatorToOutput()
204 auto hessians = hessians_t->flat_outer_dims<float>(); in SerializeTensorAccumulatorToOutput()
277 const auto& gradients = gradients_t.flat_outer_dims<float>(); in AddToTensorAccumulator()
279 const auto& hessians = hessians_t.flat_outer_dims<float>(); in AddToTensorAccumulator()
/external/tensorflow/tensorflow/core/kernels/data/
Dmap_and_batch_dataset_op.cc229 auto output_t = output->flat_outer_dims<type>(); \ in CopyPartialBatch()
230 auto value_t = value.flat_outer_dims<type>(); \ in CopyPartialBatch()